Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
| 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:)
|