Update src/ai_processor.py

src/ai_processor.py

@@ -24,9 +24,9 @@ if not os.path.exists(UPLOADS_DIR):
     logging.info(f"Created uploads directory: {UPLOADS_DIR}")
 
 HF_TOKEN = os.getenv("HF_TOKEN")
-YOLO_MODEL_PATH = "best.pt"
-SEG_MODEL_PATH = "segmentation_model.h5"
-GUIDELINE_PDFS = ["eHealth in Wound Care.pdf", "IWGDF Guideline.pdf", "evaluation.pdf"]
+YOLO_MODEL_PATH = "src/best.pt"
+SEG_MODEL_PATH = "src/segmentation_model.h5"
+GUIDELINE_PDFS = ["src/eHealth in Wound Care.pdf", "src/IWGDF Guideline.pdf", "src/evaluation.pdf"]
 DATASET_ID = "SmartHeal/wound-image-uploads"
 MAX_NEW_TOKENS = 2048
 PIXELS_PER_CM = 38