Update app.py

app.py

@@ -18,17 +18,21 @@ model = model.to(device)
 
 def load_img (filename):
     img = Image.open(filename).convert("RGB")
-    img_tensor = pil_to_tensor(img)
-    return img_tensor
+    return img
 
 def process_img(image):
     y = img.to(device)
 
     with torch.no_grad():
         result = model(y)
-            
-    return Image.fromarray(result)
 
+        if len(result[0].boxes.boxes)>0:
+        vis = plots.plot_detection(img,boxes=result[0].boxes.boxes,
+                     class_names=class_names, min_score=0.2)
+        else:
+            vis = img
+    return vis
+    
 title = "Efficient Hazy Vehicle Detection ✏️[] 🤗"
 description = ''' ## [Efficient Hazy Vehicle Detection](https://github.com/cidautai)
 [Paula Garrido Mellado](https://github.com/paugar5)