Spaces:
Runtime error
Runtime error
Commit
·
1969b0a
1
Parent(s):
85d8257
Update app.py
Browse files
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()
|