update

app.py

@@ -1,4 +1,3 @@
-#Falah with Gradio
 import gradio as gr
 from transformers import pipeline
 from PIL import Image, ImageDraw
@@ -18,14 +17,17 @@ def detect_and_visualize_objects(image):
 
     # Draw bounding boxes and labels on the image
     draw = ImageDraw.Draw(image)
-    for prediction in predictions:
-        box = prediction["box"]
-        label = prediction["label"]
-        score = prediction["score"]
-
-        xmin, ymin, xmax, ymax = box.values()
-        draw.rectangle((xmin, ymin, xmax, ymax), outline="red", width=1)
-        draw.text((xmin, ymin), f"{label}: {round(score, 2)}", fill="white")
+    if len(predictions) == 0:
+        draw.text((10, 10), "Object not found in image", fill="red")
+    else:
+        for prediction in predictions:
+            box = prediction["box"]
+            label = prediction["label"]
+            score = prediction["score"]
+
+            xmin, ymin, xmax, ymax = box.values()
+            draw.rectangle((xmin, ymin, xmax, ymax), outline="red", width=1)
+            draw.text((xmin, ymin), f"{label}: {round(score, 2)}", fill="white")
 
     # Return the annotated image
     return image
@@ -38,8 +40,8 @@ iface = gr.Interface(
     fn=detect_and_visualize_objects,
     inputs=image_input,
     outputs=image_output,
-    title="Space and War Missile Detection System ",
-    description="Detect objects in an image using a pre-trained model and visualize the results.",
+    title="Space and War Missile Detection System",
+    description="Detect objects in an image using a pre-trained model and visualize the results."
 )
 
 # Launch the Gradio interface