JaredBailey commited on
Commit
3bb67ac
·
verified ·
1 Parent(s): 568cc07

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -69,8 +69,9 @@ model_loaded.load_state_dict(torch.load(model_path, map_location='cpu'))
69
  # Set the model to evaluation mode
70
  model_loaded.eval()
71
 
 
 
72
  if result not None:
73
- result = predict_image(image_path=img, model=model_loaded)
74
- st.write(result)
75
  st.write(:chipmunk:)
76
  st.write(:bird:)
 
69
  # Set the model to evaluation mode
70
  model_loaded.eval()
71
 
72
+ result = predict_image(image_path=img, model=model_loaded)
73
+
74
  if result not None:
75
+ st.write(result)
 
76
  st.write(:chipmunk:)
77
  st.write(:bird:)