Spaces:

user1729
/

cancer_classify_extract-api

Sleeping

user1729 commited on Jun 21, 2025

Commit

1a15d27

1 Parent(s): e8fb365

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("uuser1729/BiomedBERT-cancer-bert-classifier-v1.0")
     logger.info("Loading extraction model...")
     extraction_pipeline = CancerExtractor()

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