Update app.py

app.py

@@ -16,20 +16,17 @@ def detect(files):
     pred_ct = res["PRED_CT"],
     geo_tag_url =  res["GEO_TAG_URL"]
     PRED.append(pred_ct)
-    return f"Predicted"
+    return f"Predicted: {PRED}"
 
 with gr.Blocks() as demo:
     with gr.Row():
-        output=gr.Image()
+        output=gr.File()
     with gr.Row():
         btn = gr.UploadButton(
             label="Upload Image",
-            file_types = ['.jpg','.jpeg'],
-            file_count = "multiple")
-        btn.upload(fn=detect, inputs=btn)
-        with gr.Column(scale=1, min_width=600):
-            gr.Markdown(f"Output here")
-            if PRED is not None:
-                gr.Markdown(f"Predicted: {PRED}")
+            file_types=[".jpg",".jpeg"],
+            file_count="multiple")
+        btn.upload(fn=detect, inputs=btn, outputs=gr.Label())
+# put gr.Label() in upload(outputs=gr.Label())
 
 demo.launch()