Fix: Enable actual model prediction instead of fallback function

main.py

@@ -6,9 +6,14 @@ from fastapi.responses import JSONResponse
 from fastapi.middleware.cors import CORSMiddleware
 
 # Import the prediction function
+import sys
+sys.path.append('/code')
+
 try:
     from app.predict import predict_image
-except ImportError:
+    print("Successfully imported predict_image function")
+except ImportError as e:
+    print(f"Failed to import predict_image: {e}")
     # Fallback prediction function for testing
     def predict_image(image_path):
         return "test_condition", 0.85, [{"label": "test_condition", "confidence": 0.85}]

requirements.txt

@@ -3,4 +3,5 @@ uvicorn[standard]
 python-multipart
 numpy
 Pillow
-tensorflow==2.10.0
+tensorflow==2.10.0
+huggingface_hub