added type pil

app.py

@@ -7,7 +7,6 @@ model = YOLO('hotspot_detector.pt')
 
 def detect_hotspots(image):
     result = model(image)
-    image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
     
     for r in result:
         im_array = r.plot()
@@ -15,5 +14,5 @@ def detect_hotspots(image):
         return Image.fromarray(im_array)
 
 
-demo = gr.Interface(fn=detect_hotspots, inputs=gr.Image(), outputs="image", title="YOLO Object Detection")
+demo = gr.Interface(fn=detect_hotspots, inputs=gr.Image(type='pil'), outputs="image", title="YOLO Object Detection")
 demo.launch()