PinLing commited on
Commit
f07d3b0
1 Parent(s): 612a975

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ def predict(img):
13
  # st.write(learn_inf.predict(img))
14
 
15
  f"""
16
- ## 牠 **{'是 ' if pred == 'mi' else '不是'}** 海獅的兄弟姐妹.
17
  ### Rediction result: {pred}
18
  ### Probability of {pred}: {probs[key].item()*100: .2f}%
19
  """
 
13
  # st.write(learn_inf.predict(img))
14
 
15
  f"""
16
+ ## 牠 **{'是 ' if probs[key].item()*100 >= 95 else '不是'}** 海獅的兄弟姐妹.
17
  ### Rediction result: {pred}
18
  ### Probability of {pred}: {probs[key].item()*100: .2f}%
19
  """