Spaces:

user1729
/

cancer_classify_extract-api

Sleeping

user1729 commited on Jun 21

Commit

e8fb365

1 Parent(s): 3ea7ed7

renamed classifier

Files changed (1) hide show

app/main.py CHANGED Viewed

@@ -30,7 +30,7 @@ class BatchResponse(BaseModel):
 # Initialize models
 try:
     logger.info("Loading classification model...")
-    classification_pipeline = CancerClassifier("user1729/BiomedBERT-cancer-classifier-v1.0")
     logger.info("Loading extraction model...")
     extraction_pipeline = CancerExtractor()

 # Initialize models
 try:
     logger.info("Loading classification model...")
+    classification_pipeline = CancerClassifier("uuser1729/BiomedBERT-cancer-bert-classifier-v1.0")
     logger.info("Loading extraction model...")
     extraction_pipeline = CancerExtractor()