Update main.py

app/main.py

@@ -30,7 +30,7 @@ class BatchResponse(BaseModel):
 # Initialize models
 try:
     logger.info("Loading classification model...")
-    classification_pipeline = CancerClassifier("models/fine_tuned")
+    classification_pipeline = CancerClassifier("user1729/BiomedBERT-cancer-classifier-v1.0")
     
     logger.info("Loading extraction model...")
     extraction_pipeline = CancerExtractor()
@@ -163,4 +163,4 @@ async def health_check():
 
 if __name__ == "__main__":
     import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=8000)
+    uvicorn.run(app, host="0.0.0.0", port=8000)