Update yolo.py

yolo.py

@@ -6,7 +6,7 @@ from datetime import datetime, timedelta
 from ultralytics import YOLO
 import cv2
 import torch
-from gradio import Interface, inputs, outputs
+from gradio import Interface  # Use 'gr' as namespace for inputs and outputs
 from io import BytesIO
 import threading
 
@@ -103,7 +103,8 @@ class FractureDetector:
 
         # Return the image in BytesIO format for Gradio
         img_byte_arr = BytesIO()
-        cv2.imwrite(img_byte_arr, img_viz, [int(cv2.IMWRITE_JPEG_QUALITY), 90])
+        _, img_encoded = cv2.imencode('.jpg', img_viz, [int(cv2.IMWRITE_JPEG_QUALITY), 90])
+        img_byte_arr.write(img_encoded)
         img_byte_arr.seek(0)  # Go to the beginning of the BytesIO buffer
         return img_byte_arr.getvalue()  # Return byte content of the image
 
@@ -131,8 +132,8 @@ def detect_fractures_in_image(image):
 # Define Gradio interface
 iface = Interface(
     fn=detect_fractures_in_image,
-    inputs=inputs.Image(type="numpy"),
-    outputs=outputs.JSON(),
+    inputs=gr.Image(type="numpy"),  # Use 'gr' to access Image class
+    outputs=gr.JSON(),  # Use 'gr' to access JSON class
     title="Fracture Detection using YOLO",
     description="Upload an X-ray image to detect fractures."
 )