Update app.py

app.py

@@ -16,18 +16,20 @@ def detect(files):
     pred_ct = res["PRED_CT"],
     geo_tag_url =  res["GEO_TAG_URL"]
     PRED.append(pred_ct)
-    return f"Predicted: {PRED}"
+    
+    print(res["PRED_CT"])
+    return res["PRED_CT"]
 
 with gr.Blocks() as demo:
+    gr.Markdown("# DetectIt")
     with gr.Row():
-        output=gr.File()
-    with gr.Row():
-        # text = gr.Textbox(show_label=False, elem_id="result-textarea")
-        btn = gr.UploadButton(
+        image = gr.UploadButton(
             label="Upload Image",
             file_types=[".jpg",".jpeg"],
             file_count="multiple")
-        btn.upload(fn=detect, inputs=btn, outputs=[gr.Textbox(show_label=False, elem_id="result-textarea")], api_name="predict")
-# put gr.Label() in upload(outputs=gr.Label())
 
-demo.launch()
+        btn = gr.Button("Go")
+    text = gr.Textbox(show_label=False, elem_id="result-textarea")
+    btn.click(detect, inputs=[image], outputs=[text], api_name="predict")
+
+demo.launch(share=True)