updated app.py

app.py

@@ -144,12 +144,15 @@ if uploaded_file is not None:
                 progress_bar.progress(i)
                 status_text.text(f"Preprocessing: {i}%")
 
-            main(cfg, input_image, output_json_name, output_image_name)
-            print("✅ Model run complete.")
 
-            # Prepare output paths
             output_json_path = os.path.join(JSON_DIR, output_json_name)
             output_image_path = os.path.join(JSON_DIR, output_image_name)
+            main(cfg, input_image, output_json_path, output_image_path)
+            print("✅ Model run complete.")
+            # main(cfg, input_image, output_json_name, output_image_name)
+            # # Prepare output paths
+            # output_json_path = os.path.join(JSON_DIR, output_json_name)
+            # output_image_path = os.path.join(JSON_DIR, output_image_name)
 
             while not os.path.exists(output_json_path):
                 print("Waiting for JSON output...")