vardaan123 commited on
Commit
e6a3d86
·
1 Parent(s): bef393e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -53,7 +53,7 @@ def evaluate(img):
53
  pred_label = target_list[y_pred].item()
54
  species_label = overall_id_to_name[str(id2entity[pred_label])]
55
 
56
- return species_label
57
 
58
  # Gradio interface
59
  species_model = gr.Interface(
 
53
  pred_label = target_list[y_pred].item()
54
  species_label = overall_id_to_name[str(id2entity[pred_label])]
55
 
56
+ return {species_label:1.0}
57
 
58
  # Gradio interface
59
  species_model = gr.Interface(