Update app.py

app.py

@@ -1,3 +1,4 @@
+
 import io
 import sys
 import time
@@ -12,6 +13,7 @@ from models.rice_leaf_validator import RiceLeafValidator, is_rice_leaf
 from utils.image_processing import process_image
 import requests
 import json
+import os
 from typing import Dict, Any
 
 # Configure logging
@@ -247,8 +249,8 @@ async def validate_rice_leaf_image(file: UploadFile = File(...)):
                 raise HTTPException(status_code=500, detail=f"Failed to load models: {last_error}")
 
     try:
-        # Save uploaded file temporarily
-        temp_file_path = f"temp_{file.filename}"
+        # Save uploaded file temporarily in /tmp (Hugging Face Spaces requirement)
+        temp_file_path = os.path.join("/tmp", f"temp_{file.filename}")
         with open(temp_file_path, "wb") as buffer:
             buffer.write(await file.read())
 
@@ -256,14 +258,12 @@ async def validate_rice_leaf_image(file: UploadFile = File(...)):
         is_rice, confidence = is_rice_leaf(temp_file_path, leaf_validator_model, device)
 
         # Remove temporary file
-        import os
         os.remove(temp_file_path)
 
         return {
             "is_rice_leaf": is_rice,
             "confidence": f"{confidence * 100:.2f}%"
         }
-
     except Exception as e:
         logging.error(f"Error during leaf validation: {e}")
         raise HTTPException(status_code=500, detail=f"Validation failed: {str(e)}")
@@ -285,8 +285,9 @@ async def predict_disease(file: UploadFile = File(...), use_ai_recommendation: b
         # Read image file
         start_time = time.time()
 
-        # Save uploaded file temporarily for validation
-        temp_file_path = f"temp_{file.filename}"
+        # Save uploaded file temporarily for validation in /tmp (Hugging Face Spaces requirement)
+        import os
+        temp_file_path = os.path.join("/tmp", f"temp_{file.filename}")
         contents = await file.read()
         with open(temp_file_path, "wb") as buffer:
             buffer.write(contents)