updating docker

app.py

@@ -37,14 +37,15 @@ os.makedirs(OUTPUT_DIR, exist_ok=True)
 if not os.path.exists(model_path):
     print("🚀 Model file not found! Downloading...")
     try:
-        gdown.download(GDRIVE_URL, model_path, quiet=False)
+        # gdown.download(GDRIVE_URL, model_path, quiet=False)
+        gdown.download(GDRIVE_URL, model_path, quiet=False, use_cookies=False)
         print("✅ Model downloaded successfully.")
     except Exception as e:
         print(f"❌ Failed to download model: {e}")
 
 # Import model
 sys.path.append(MODEL_DIR)
-from rcnn_run import main, write_config
+from rcnn_model.scripts.rcnn_run import main, write_config
 
 cfg = write_config()
 
@@ -55,14 +56,16 @@ cfg = write_config()
 def predict(uploaded_file):
     if uploaded_file is None:
         return None, None, "No file uploaded."
-
+    uploaded_path = uploaded_file
+    input_filename = os.path.basename(uploaded_path)
+    print(f"✅ Image received at {uploaded_path}")
     # Save uploaded image
-    input_bytes = uploaded_file.read()
-    img = Image.open(BytesIO(input_bytes)).convert("RGB")
-    input_filename = uploaded_file.name
-    uploaded_path = os.path.join(UPLOAD_DIR, input_filename)
-    img.save(uploaded_path)
-    print(f"✅ Image saved to {uploaded_path}")
+    # input_bytes = uploaded_file.read()
+    # img = Image.open(BytesIO(input_bytes)).convert("RGB")
+    # input_filename = uploaded_file.name
+    # uploaded_path = os.path.join(UPLOAD_DIR, input_filename)
+    # img.save(uploaded_path)
+    # print(f"✅ Image saved to {uploaded_path}")
 
     # Prepare output paths
     output_json_name = input_filename.replace(".png", "_result.json").replace(".jpg", "_result.json").replace(".jpeg", "_result.json")
@@ -92,7 +95,8 @@ with gr.Blocks() as demo:
 
     with gr.Row():
         with gr.Column():
-            uploaded_file = gr.File(label="Upload your Floorplan Image", type="file")
+            uploaded_file = gr.File(label="Upload your Floorplan Image", type="filepath")
+            # uploaded_file = gr.File(label="Upload your Floorplan Image", type="file")
             run_button = gr.Button("Run Vectorizer 🔥")
 
         with gr.Column():