fixing app

app.py

@@ -15,14 +15,15 @@ def init_model(model_id: str):
     }
     
     if model_id in MODEL_OPTIONS:
-        os.makedirs("models", exist_ok=True)
+        print(MODEL_OPTIONS[model_id])
         path = hf_hub_download(
             repo_id="biglam/medieval-manuscript-yolov11",
-            filename=MODEL_OPTIONS[model_id]
+            filename=MODEL_OPTIONS[model_id],
         )
-        local_path = os.path.join("models", path)
+        print(path)
         # Initialize and return model
         model = YOLO(path)
+        print("Model initialized")
         return model
     else:
         raise ValueError(f"Model {model_id} not found")
@@ -39,15 +40,16 @@ async def predict(image: UploadFile,
                  conf: float = 0.25,
                  iou: float = 0.7
                  ):
+    print(model_id, conf, iou)
     # Initialize model for each request
     model = init_model(model_id)
     
     # Open image from uploaded file
     image = Image.open(image.file)
-    
+    print("Image opened")
     # Run inference with the PIL Image
     results = model.predict(source=image, conf=conf, iou=iou)
-    
+    print("Inference done")
     # Extract detection results
     result = results[0]
     detections = []

test.py

@@ -21,6 +21,10 @@ with open(image_path, 'rb') as f:
     response = requests.post(local_app + ':7860/predict', 
                            files=files,
                            params=params)
+    print(response.json())
+    response = requests.post(hf_app + ':7860/predict', 
+                           files=files,
+                           params=params)
 
 # Print the results
 print(response.json())