Update app.py

app.py

@@ -34,12 +34,9 @@ def detect_objects(image):
             draw.rectangle(box, outline="red", width=3)
             draw.text((box[0], box[1]), f"{label_name}: {round(score.item(), 3)}", fill="red")
     
-    # แปลงภาพเป็นรูปแบบที่สามารถแสดงผลได้ใน Gradio
-    output_image = io.BytesIO()
-    image.save(output_image, format='PNG')
-    output_image.seek(0)
-    
-    return output_image
+    # แปลงภาพกลับเป็นรูปแบบที่ Gradio สามารถแสดงได้
+    pil_image = Image.fromarray(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))
+    return pil_imag
 
 # สร้างอินเตอร์เฟซด้วย Gradio
 interface = gr.Interface(