updated app.py

app.py

@@ -6,6 +6,7 @@ import os
 import sys
 import shutil
 import gdown
+from io import BytesIO
 
 # ==================================
 # SETUP
@@ -50,7 +51,7 @@ if not os.path.exists(model_path):
 # ==================================
 
 sys.path.append(MODEL_DIR)
-from rcnn_run import main, write_config
+from rcnn_model.scripts.rcnn_run import main, write_config
 
 # ==================================
 # PAGE CONFIG
@@ -95,6 +96,9 @@ col1, col2 = st.columns([1, 2])
 if uploaded_file is not None:
     print("📤 File Uploaded:", uploaded_file.name)
 
+    image_bytes = uploaded_file.read()
+    img = Image.open(BytesIO(image_bytes)).convert("RGB")
+
     uploaded_path = os.path.join(UPLOAD_DIR, uploaded_file.name)
     with open(uploaded_path, "wb") as f:
         f.write(uploaded_file.getbuffer())