Update app.py

app.py

@@ -17,17 +17,17 @@ def image_mod(image):
     yhat = model.predict(np.expand_dims(resize,0))
     display = np.argmax(yhat)
     display = str(display)
-    if display == "0":
-        message = "Rainy" # Jida,_Zhuhai,_rainy_day.jpg
-    if display == "1":
-        message = "Sunny" # Morning_fog_-_Flickr_-_tmoravec.jpg
-    if display == "2":
-        message = "Foggy" # Stuyvesant_Fish_House_25_E78_St_cloudy_jeh.jpg
-    if display == "3":
-        message = "Cloudy" #Snow_on_Branches,_Beechview,_2020-12-17,_01.jpg
-    if display == "4":
-        message = "Snowy" # Daedalus_000355_171913_516869_4578_(36155269413).jpg
-    return message
+    #if display == "0":
+    #    message = "Rainy" # Jida,_Zhuhai,_rainy_day.jpg
+    #if display == "1":
+    #    message = "Sunny" # Morning_fog_-_Flickr_-_tmoravec.jpg
+    #if display == "2":
+    #    message = "Foggy" # Stuyvesant_Fish_House_25_E78_St_cloudy_jeh.jpg
+    #if display == "3":
+    #    message = "Cloudy" #Snow_on_Branches,_Beechview,_2020-12-17,_01.jpg
+    #if display == "4":
+    #    message = "Snowy" # Daedalus_000355_171913_516869_4578_(36155269413).jpg
+    return display
 
 gr.Interface(fn=image_mod,
              inputs=gr.Image(shape=(256, 256)),