Update app.py

app.py

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