add app

app.py

@@ -7,16 +7,6 @@ import torch
 
 DEFAULT_MODEL_URL = "https://github.com/luisarizmendi/ai-apps/raw/refs/heads/main/models/luisarizmendi/object-detector-hardhat-or-hat/object-detector-hardhat-or-hat.pt"
 
-def load_model(model_input):
-    model = YOLO(model_input)
-    if torch.cuda.is_available():
-        model.to('cuda')
-        print("Using GPU for inference")
-    else:
-        print("Using CPU for inference")
-
-    return model
-
 def detect_objects_in_files(model_input, files):
     """
     Processes uploaded images for object detection.
@@ -24,7 +14,12 @@ def detect_objects_in_files(model_input, files):
     if not files:
         return "No files uploaded.", []
 
-    model = load_model(model_input)
+    model = YOLO(str(model_input))
+    if torch.cuda.is_available():
+        model.to('cuda')
+        print("Using GPU for inference")
+    else:
+        print("Using CPU for inference")
 
     results_images = []
     for file in files: