Spaces:

uoda5t
/

test_gradio

Runtime error

John Smith commited on Nov 7, 2023

Commit

ade45a8

1 Parent(s): b00effd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ def image_mod(image_mod):
     resize = tf.image.resize(img, (256, 256))
     plt.imshow(resize.numpy().astype(int))
     yhat = model.predict(np.expand_dims(resize,0))
-    display = yhat
     display = str(display)
     return display

     resize = tf.image.resize(img, (256, 256))
     plt.imshow(resize.numpy().astype(int))
     yhat = model.predict(np.expand_dims(resize,0))
+    display = np.argmax(yhat)
     display = str(display)
     return display