folder-classifier 0.2.0__tar.gz → 0.2.1__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (14) hide show
  1. {folder_classifier-0.2.0 → folder_classifier-0.2.1}/PKG-INFO +1 -1
  2. {folder_classifier-0.2.0 → folder_classifier-0.2.1}/folder_classifier/classifier.py +9 -6
  3. {folder_classifier-0.2.0 → folder_classifier-0.2.1}/folder_classifier.egg-info/PKG-INFO +1 -1
  4. {folder_classifier-0.2.0 → folder_classifier-0.2.1}/setup.cfg +1 -1
  5. {folder_classifier-0.2.0 → folder_classifier-0.2.1}/README.md +0 -0
  6. {folder_classifier-0.2.0 → folder_classifier-0.2.1}/folder_classifier/__init__.py +0 -0
  7. {folder_classifier-0.2.0 → folder_classifier-0.2.1}/folder_classifier/app.py +0 -0
  8. {folder_classifier-0.2.0 → folder_classifier-0.2.1}/folder_classifier/deploy.py +0 -0
  9. {folder_classifier-0.2.0 → folder_classifier-0.2.1}/folder_classifier/dto.py +0 -0
  10. {folder_classifier-0.2.0 → folder_classifier-0.2.1}/folder_classifier/util.py +0 -0
  11. {folder_classifier-0.2.0 → folder_classifier-0.2.1}/folder_classifier.egg-info/SOURCES.txt +0 -0
  12. {folder_classifier-0.2.0 → folder_classifier-0.2.1}/folder_classifier.egg-info/dependency_links.txt +0 -0
  13. {folder_classifier-0.2.0 → folder_classifier-0.2.1}/folder_classifier.egg-info/top_level.txt +0 -0
  14. {folder_classifier-0.2.0 → folder_classifier-0.2.1}/pyproject.toml +0 -0
{folder_classifier-0.2.0 → folder_classifier-0.2.1}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: folder-classifier
3
- Version: 0.2.0
3
+ Version: 0.2.1
4
4
  Summary: Deploy folder classifier API to a Ray cluster
5
5
  Author: Crispin Almodovar
6
6
  Author-email:
{folder_classifier-0.2.0 → folder_classifier-0.2.1}/folder_classifier/classifier.py RENAMED
@@ -6,16 +6,19 @@ from transformers import pipeline
6
6
 
7
7
  from folder_classifier.dto import Listing
8
8
 
9
- classifier = pipeline(
10
- "zero-shot-classification",
11
- model="MoritzLaurer/ModernBERT-large-zeroshot-v2.0",
12
- torch_dtype=torch.bfloat16,
13
- device="cuda"
14
- )
9
+ classifier = None
15
10
 
16
11
  candidate_labels = ["legal_matter", "other"]
17
12
 
18
13
  def predict(listing: Listing) -> Tuple[str, float]:
14
+ global classifier
15
+ if classifier is None:
16
+ classifier = pipeline(
17
+ "zero-shot-classification",
18
+ model="MoritzLaurer/ModernBERT-large-zeroshot-v2.0",
19
+ torch_dtype=torch.bfloat16,
20
+ device="cuda"
21
+ )
19
22
  text = "\n".join(listing.items)
20
23
  hypothesis_template = "This list of files is about {}"
21
24
  prediction = classifier(
{folder_classifier-0.2.0 → folder_classifier-0.2.1}/folder_classifier.egg-info/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: folder-classifier
3
- Version: 0.2.0
3
+ Version: 0.2.1
4
4
  Summary: Deploy folder classifier API to a Ray cluster
5
5
  Author: Crispin Almodovar
6
6
  Author-email:
{folder_classifier-0.2.0 → folder_classifier-0.2.1}/setup.cfg RENAMED
@@ -1,6 +1,6 @@
1
1
  [metadata]
2
2
  name = folder-classifier
3
- version = 0.2.0
3
+ version = 0.2.1
4
4
  author = Crispin Almodovar
5
5
  author_email =
6
6
  description = Deploy folder classifier API to a Ray cluster