Update app.py

app.py

@@ -287,7 +287,17 @@ class RiceLeafValidator:
         from torchvision.models import resnet18
         
         model = resnet18(weights=None)
-        model.fc = nn.Linear(model.fc.in_features, 2)  # Binary classification
+        
+        # The validator was trained with a Sequential FC head:
+        # fc.0 = Linear(512, hidden), fc.1 = ReLU, fc.2 = Dropout, fc.3 = Linear(hidden, 2)
+        # We need to match this architecture exactly
+        in_features = model.fc.in_features  # 512 for ResNet18
+        model.fc = nn.Sequential(
+            nn.Linear(in_features, 256),
+            nn.ReLU(),
+            nn.Dropout(0.5),
+            nn.Linear(256, 2)  # Binary classification: rice / not rice
+        )
         
         state_dict = torch.load(model_path, map_location=self.device, weights_only=True)
         model.load_state_dict(state_dict)
@@ -613,6 +623,5 @@ async def predict_disease(file: UploadFile = File(...), use_ai_recommendation: b
 
 if __name__ == "__main__":
     import uvicorn
-    # Hugging Face Spaces uses port 7860
-    port = int(os.environ.get("PORT", 7860))
-    uvicorn.run("app:app", host="0.0.0.0", port=port, reload=False)
+    # Production settings: no hot reload, proper host binding
+    uvicorn.run("app:app", host="0.0.0.0", port=8000, reload=False)