user1729 commited on
Commit
34b213c
·
unverified ·
1 Parent(s): b462bc5

Update model.py

Browse files
Files changed (1) hide show
  1. 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)