Update app.py

app.py

@@ -15,9 +15,9 @@ 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))
-    str(yhat)
     #display = np.argmax(yhat)
-    #display = str(display)
+    display = np(yhat)
+    display = str(display)
     #if display == "0":
     #    message = "Rainy" # Jida,_Zhuhai,_rainy_day.jpg
     #if display == "1":
@@ -28,7 +28,7 @@ def image_mod(image):
     #    message = "Cloudy" #Snow_on_Branches,_Beechview,_2020-12-17,_01.jpg
     #if display == "4":
     #    message = "Snowy" # Daedalus_000355_171913_516869_4578_(36155269413).jpg
-    return yhat
+    return display
 
 gr.Interface(fn=image_mod,
              inputs=gr.Image(shape=(256, 256)),