Jamshaid89 commited on
Commit
2566994
1 Parent(s): 31649fa

Corrected predicted label

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ def predict_image(image):
19
  else:
20
  predicted_label = "Dog"
21
 
22
- predicted_label = f"{predicted_label} with prediction : {pred}"
23
  # Return the predicted label
24
  return predicted_label
25
 
 
19
  else:
20
  predicted_label = "Dog"
21
 
22
+ predicted_label = f"{predicted_label} with prediction : {pred[0][0]}"
23
  # Return the predicted label
24
  return predicted_label
25