anuragshas commited on
Commit
1969b0a
1 Parent(s): 85d8257

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -53,7 +53,7 @@ def classify_image(img_path):
53
  img = tf.expand_dims(img, axis=0)
54
  preds = prediction_model.predict(img)
55
  pred_text = decode_batch_predictions(preds)
56
- return pred_text
57
 
58
  image = gr.inputs.Image(type='filepath')
59
  text = gr.outputs.Textbox()
 
53
  img = tf.expand_dims(img, axis=0)
54
  preds = prediction_model.predict(img)
55
  pred_text = decode_batch_predictions(preds)
56
+ return pred_text[0]
57
 
58
  image = gr.inputs.Image(type='filepath')
59
  text = gr.outputs.Textbox()