Update app.py

app.py

@@ -9,18 +9,20 @@ torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/raw/master
 # Model
 model = torch.hub.load('ultralytics/yolov5', 'yolov5s')  # force_reload=True to update
 
-
+os.mkdir("out")
 def yolo(im, size=640):
     g = (size / max(im.size))  # gain
     im = im.resize((int(x * g) for x in im.size), Image.ANTIALIAS)  # resize
 
     results = model(im)  # inference
     results.render()  # updates results.imgs with boxes and labels
-    return Image.fromarray(results.imgs[0])
+    results.save()
+    os.system("ls")
+    return "out.png"
 
 
 inputs = gr.inputs.Image(type='pil', label="Original Image")
-outputs = gr.outputs.Image(type="pil", label="Output Image")
+outputs = gr.outputs.Image(type="file", label="Output Image")
 
 title = "YOLOv5"
 description = "YOLOv5 Gradio demo for object detection. Upload an image or click an example image to use."