fixing model download path from remote

app.py

@@ -3,32 +3,38 @@ from PIL import Image
 import gradio as gr
 from huggingface_hub import snapshot_download
 import os
+import shutil
 
-# Get model path once at startup
 def initialize_model_path():
     is_huggingface = os.environ.get("SPACE_ID") is not None
     if is_huggingface:
         REPO_ID = "mail2kandan/guitar_watch_openvino_model"
         print("Downloading model from Hugging Face...")
         
-        # Create a model directory
-        model_dir = 'openvino_model'
+        # Create a temp directory for the model
+        model_dir = os.path.join(os.getcwd(), 'guitar_openvino_model')
         os.makedirs(model_dir, exist_ok=True)
         
-        # Download both files
-        snapshot_download(
-            repo_id=REPO_ID,
-            allow_patterns=["*.xml", "*.bin"],
-            local_dir=model_dir
-        )
+        # Download complete model directory
+        downloaded_path = snapshot_download(repo_id=REPO_ID)
+        
+        # Copy files to our model directory
+        for ext in ['.xml', '.bin']:
+            src = os.path.join(downloaded_path, f'best{ext}')
+            dst = os.path.join(model_dir, f'best{ext}')
+            if os.path.exists(src):
+                shutil.copy2(src, dst)
+                print(f"Copied {src} to {dst}")
+            else:
+                raise Exception(f"Required model file best{ext} not found in repository")
         
-        # Verify files exist
+        # Verify both files exist in the new directory
         if os.path.exists(os.path.join(model_dir, 'best.xml')) and \
            os.path.exists(os.path.join(model_dir, 'best.bin')):
-            print(f"Model files downloaded to: {model_dir}")
+            print(f"Using model directory: {model_dir}")
             return model_dir
         else:
-            raise Exception("Model files not found after download")
+            raise Exception("Model files not found in target directory after copying")
     else:
         local_path = r'C:\Manikandan\NYP\ITI107\Assignment_draft\output_models\best_openvino_model'
         print(f"Using local model from: {local_path}")
@@ -41,12 +47,21 @@ def predict(pilimg, conf_threshold, iou_threshold):
     try:
         if pilimg is None:
             return None
+            
+        print(f"Loading model from: {MODEL_PATH}")
+        print(f"Model path exists: {os.path.exists(MODEL_PATH)}")
+        if os.path.isfile(MODEL_PATH):
+            print(f"Found these files in directory: {os.listdir(os.path.dirname(MODEL_PATH))}")
+        
         model = YOLO(MODEL_PATH, task='detect')
         result = model.predict(pilimg, conf=conf_threshold, iou=iou_threshold)
         img_bgr = result[0].plot()
         return Image.fromarray(img_bgr[..., ::-1])
     except Exception as e:
         print(f"Error during prediction: {str(e)}")
+        # Print full error traceback for debugging
+        import traceback
+        print(traceback.format_exc())
         return None
 
 with gr.Blocks() as demo:
@@ -54,7 +69,7 @@ with gr.Blocks() as demo:
     gr.HTML("""
         <div style="text-align: center; margin-bottom: 20px; padding: 20px; background-color: #f7f7f7;">
             <h1 style="font-size: 2.5em; color: #2196F3; margin-bottom: 10px;">Guitar & Watch Detection</h1>
-            <p style="font-size: 1.2em; color: #666;">Powered by YOLOv8 </p>
+            <p style="font-size: 1.2em; color: #666;">Powered by YOLOv8</p>
         </div>
     """)
     
@@ -142,10 +157,11 @@ with gr.Blocks() as demo:
     # Add custom CSS to resize the hero image
     demo.css = """
     #hero-image img {
-        width: 560px; /* Adjust the width to make the image smaller */
+        width: 560px;
         height: auto;
     }
     """
 
 if __name__ == "__main__":
-    demo.launch(share=True)
+    is_huggingface = os.environ.get("SPACE_ID") is not None
+    demo.launch(share=not is_huggingface)