Spaces:

lakiet
/

tomato_plants_diseases

Runtime error

lakiet commited on Jun 7, 2023

Commit

b7edbf3

•

1 Parent(s): caa2a14

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -33,7 +33,7 @@ def predict_gradio(img):
     #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), map_location=torch.device("cpu"))
     loaded_model_2.eval()
     with torch.inference_mode():

     #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, map_location=torch.device("cpu")))
     loaded_model_2.eval()
     with torch.inference_mode():