Update app.py

app.py

@@ -18,6 +18,7 @@ from utils.general import check_img_size, check_requirements, check_imshow, non_
     scale_coords, xyxy2xywh, strip_optimizer, set_logging, increment_path
 from utils.plots import plot_one_box
 from utils.torch_utils import select_device, load_classifier, time_synchronized, TracedModel
+from PIL import Image
 
 parser = argparse.ArgumentParser()
 parser.add_argument('--weights', nargs='+', type=str, default='yolov7.pt', help='model.pt path(s)')
@@ -179,11 +180,11 @@ def detect(img):
 
     print(f'Done. ({time.time() - t0:.3f}s)')
 
-    return im0
+    return Image.fromarray(im0)
 
 with torch.no_grad():
     if opt.update:  # update all models (to fix SourceChangeWarning)
         for opt.weights in ['yolov7.pt']:
             strip_optimizer(opt.weights)
    
-gr.Interface(detect,gr.Image(type="pil"), gr.Image(type="numpy")).launch()
+gr.Interface(detect,gr.Image(type="pil"), gr.Image(type="pil")).launch()