Update model.py
Browse files- app/model.py +1 -2
app/model.py
CHANGED
@@ -7,7 +7,6 @@ class CancerClassifier:
|
|
7 |
self.classifier = pipeline(
|
8 |
"text-classification",
|
9 |
model=model_path,
|
10 |
-
tokenizer="microsoft/BiomedNLP-BiomedBERT-base-uncased-abstract",
|
11 |
return_all_scores=True,
|
12 |
device=0 if os.environ.get("USE_GPU", "false").lower() == "true" else -1,
|
13 |
)
|
@@ -102,4 +101,4 @@ class CancerExtractor:
|
|
102 |
for word2 in text_list
|
103 |
if any(word1.lower() in word2.lower() for word1 in self.cancers)
|
104 |
]
|
105 |
-
return set(detected_cancers)
|
|
|
7 |
self.classifier = pipeline(
|
8 |
"text-classification",
|
9 |
model=model_path,
|
|
|
10 |
return_all_scores=True,
|
11 |
device=0 if os.environ.get("USE_GPU", "false").lower() == "true" else -1,
|
12 |
)
|
|
|
101 |
for word2 in text_list
|
102 |
if any(word1.lower() in word2.lower() for word1 in self.cancers)
|
103 |
]
|
104 |
+
return set(detected_cancers)
|