Spaces:

intelliarts
/

hotspot-anomaly-detection-for-solar-panels

Running

Ubuntu commited on Aug 7, 2024

Commit

56ef960

1 Parent(s): ea07644

changed image order

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ def detect_hotspots(image):
     for r in result:
         im_array = r.plot()
         # im = Image.fromarray(im_array[..., ::-1])
-        return Image.fromarray(im_array[..., ::-1])
 demo = gr.Interface(fn=detect_hotspots, inputs=gr.Image(), outputs="image", title="YOLO Object Detection")

     for r in result:
         im_array = r.plot()
         # im = Image.fromarray(im_array[..., ::-1])
+        return Image.fromarray(im_array)
 demo = gr.Interface(fn=detect_hotspots, inputs=gr.Image(), outputs="image", title="YOLO Object Detection")