Update app.py

app.py

@@ -13,7 +13,17 @@ def image_mod(image_mod):
     yhat = model.predict(np.expand_dims(resize,0))
     display = np.argmax(yhat)
     display = str(display)
-    return display
+    if display == "0":
+        message = "Rainy"
+    if display == "1":
+        message = "Foggy"
+    if display == "2":
+        message = "Cloudy"
+    if display == "3":
+        message = "Snowy"
+    if display == "4":
+        message = "Sunny"
+    return message
 
 gr.Interface(fn=image_mod,
              inputs=gr.Image(shape=(256, 256)),