JaredBailey commited on
Commit
0455e06
·
verified ·
1 Parent(s): 3c6cb52

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -72,6 +72,7 @@ model_loaded.eval()
72
  result = predict_image(image_path=img, model=model_loaded)
73
 
74
  if result is not None:
75
- st.write(result)
76
- st.write(":chipmunk:")
77
- st.write(":bird:")
 
 
72
  result = predict_image(image_path=img, model=model_loaded)
73
 
74
  if result is not None:
75
+ if result == 'bird':
76
+ st.write(":bird:" + "That's a Bird" + ":bird:")
77
+ else:
78
+ st.write(":chipmunk:" + "That's a Squirrel" + ":chipmunk:")