Refactor model loading in app.py to create a models directory and update local model path. Modify test.py to change default model ID and use a variable for the local app URL.

app.py

@@ -20,11 +20,13 @@ def init_model(model_id: str):
         "YOLOv11-XLarge": "medieval-yolov11x.pt"
     }
     if model_id in MODEL_OPTIONS:
+        os.makedirs("models", exist_ok=True)
         model_path = hf_hub_download(
             repo_id="biglam/medieval-manuscript-yolov11",
             filename=MODEL_OPTIONS[model_id]
         )
-        return YOLO(model_path)
+        local_path = os.path.join("models", model_path)
+        return YOLO(local_path)
     else:
         raise ValueError(f"Model {model_id} not found")
 

test.py

@@ -3,6 +3,8 @@ import requests
 # Path to your image
 image_path = "/Users/wjm55/yale/weaviate-test/yolov11_output/valid/images/Paris_BnF_Velins_611_00003.jpg"
 
+hf_app = "https://wjbmattingly-medieval-yolo-api.hf.space"
+local_app = "http://localhost"
 # Open the image file
 with open(image_path, 'rb') as f:
     # Create the files parameter for the POST request
@@ -10,13 +12,13 @@ with open(image_path, 'rb') as f:
     
     # Optional parameters (using defaults from the API)
     params = {
-        'model_id': 'YOLOv11-XLarge',  # default model
+        'model_id': 'YOLOv11-Nano',  # default model
         'conf': 0.25,                   # confidence threshold
         'iou': 0.7                      # IoU threshold
     }
 
     # Send POST request to the endpoint
-    response = requests.post('http://localhost:7860/predict', 
+    response = requests.post(local_app + ':7860/predict', 
                            files=files,
                            params=params)