Update app.py

app.py

@@ -15,6 +15,8 @@ def get_model_path(model_name):
 
 @spaces.GPU
 def yolo_inference(images, model_id, conf_threshold, iou_threshold, max_detection):
+    if images is None:
+    return None
     model_path = get_model_path(model_id)  # Download model
     model_type = RTDETR if 'rtdetr' in model_id.lower() else YOLO
     model = model_type(model_path)