Spaces:

lakiet
/

tomato_plants_diseases

Runtime error

lakiet commited on Jun 7, 2023

Commit

b1de449

•

1 Parent(s): 64e7e78

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -32,7 +32,7 @@ def predict_gradio(img):
     #Load model
-    model_path = r'Deployment\tomato_plants\model_2.pth'
     loaded_model_2 = custom_model(input_param=3, output_param=10)
     loaded_model_2.load_state_dict(torch.load(f=model_path))
     loaded_model_2 = loaded_model_2.to(device)

     #Load model
+    model_path = 'model_2.pth'
     loaded_model_2 = custom_model(input_param=3, output_param=10)
     loaded_model_2.load_state_dict(torch.load(f=model_path))
     loaded_model_2 = loaded_model_2.to(device)