Update app.py

app.py

@@ -11,14 +11,13 @@ def detect(files):
         save_path="./"
     )
     res = model.run()
-    img_id = res["IMG_ID"]
-    pred_lab = res["PRED_LAB"],
-    pred_ct = res["PRED_CT"],
-    geo_tag_url =  res["GEO_TAG_URL"]
-    PRED.append(pred_ct)
-    
-    print(res["PRED_CT"])
-    return res["PRED_CT"]
+    # img_id = res["IMG_ID"]
+    # pred_lab = res["PRED_LAB"],
+    # pred_ct = res["PRED_CT"],
+    # geo_tag_url =  res["GEO_TAG_URL"]
+
+    result = json.dumps(res)
+    return result
 
 with gr.Blocks() as demo:
     gr.Markdown("# DetectIt")