Spaces:

uoda5t
/

test_gradio

Runtime error

John Smith commited on Nov 7, 2023

Commit

82e76f8

1 Parent(s): 8fd9060

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,8 +15,7 @@ def image_mod(image):
     resize = tf.image.resize(img, (256, 256))
     plt.imshow(resize.numpy().astype(int))
     yhat = model.predict(np.expand_dims(resize,0))
-    if yhat == "":
-        yhat = "None"
     #display = np.argmax(yhat)
     #display = str(display)
     #if display == "0":

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