avoid output image related bug

app.py

@@ -160,7 +160,7 @@ with gr.Blocks() as app:
 
 
     button = gr.Button("Classify")
-    output_image = gr.Image()
+    #output_image = gr.Image()
 
     download_image_btn = gr.DownloadButton(label="Download Image", value=IMAGE_PATH, interactive=False)
     download_csv_btn = gr.DownloadButton(label="Download Damage List", value=CSV_PATH, interactive=False)
@@ -170,13 +170,14 @@ with gr.Blocks() as app:
     button.click(
         damage_classification,
         inputs=[image_input, cluster_threshold_input, model1_threshold_input, model2_threshold_input],
-        outputs=[output_image]
+        outputs=gr.Image(label="Output Image")
     )
 
 
 
 
 
+
 # simple interface, no title, etc
 # app = gr.Interface(damage_classification, 
 #                    inputs=[gr.Image(),