Update main.py

main.py

@@ -88,7 +88,7 @@ async def predict_plant_disease(plant_name: str, file: UploadFile = File(...)):
         raise HTTPException(status_code=400, detail="Invalid plant name")
 
     # Construct the model path
-    
+    model_path = os.path.join(MODEL_DIRECTORY, f"model_{plant_name}.keras")
     if plant_name == "Rice":
         model = load_model(model_path)
     else: