Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-import PIL.Image as Image
+import PIL.Image as Image, ImageDraw, ImageFont
 from ultralytics import YOLO, RTDETR
 import spaces
 import os
@@ -16,7 +16,20 @@ 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
+        # 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_path = get_model_path(model_id)  # Download model
     model_type = RTDETR if 'rtdetr' in model_id.lower() else YOLO
     model = model_type(model_path)