Update app.py

app.py

@@ -126,7 +126,7 @@ with gr.Blocks() as demo:
                      "day, night, dawn, dusk"], inputs=label_text)
         with gr.Row():
             with gr.Column(variant="panel"):
-                im = gr.Image(interactive=False).style(height=height)
+                im = gr.Image(interactive=False, type="pil").style(height=height)
                 with gr.Row():
                     get_btn = gr.Button("Get Random Image").style(full_width=False)
                     reclass_btn = gr.Button("Re-Classify Image").style(full_width=False)