Update app.py

app.py

@@ -5,6 +5,21 @@ import spaces
 
 @spaces.GPU
 def yolo_inference(images, model_id, conf_threshold, iou_threshold, max_detection):
+    if images is None:
+        # Create a blank image
+        width, height = 640, 480
+        blank_image = Image.new("RGB", (width, height), color="white")
+        draw = ImageDraw.Draw(blank_image)
+        message = "No image provided"
+        font = ImageFont.load_default(size=40)
+        bbox = draw.textbbox((0, 0), message, font=font)
+        text_width = bbox[2] - bbox[0]
+        text_height = bbox[3] - bbox[1]
+        text_x = (width - text_width) / 2
+        text_y = (height - text_height) / 2
+        draw.text((text_x, text_y), message, fill="black", font=font)
+        return blank_image
+    
     model = YOLO(model_id)
     results = model.predict(
         source=images,