Update app.py

app.py

@@ -24,7 +24,7 @@ def Custom_detect(img):
     parser = argparse.ArgumentParser()
     parser.add_argument('--weights', nargs='+', type=str, default=model+".pt", help='model.pt path(s)')
     parser.add_argument('--source', type=str, default='Temp_file/', help='source') 
-    parser.add_argument('--img-size', type=int, default=256, help='inference size (pixels)')
+    parser.add_argument('--img-size', type=int, default=100, help='inference size (pixels)')
     parser.add_argument('--conf-thres', type=float, default=0.25, help='object confidence threshold')
     parser.add_argument('--iou-thres', type=float, default=0.45, help='IOU threshold for NMS')
     parser.add_argument('--device', default='', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')
@@ -130,6 +130,7 @@ def Custom_detect(img):
 
             if save_img:
                 if dataset.mode == 'image':
+                    im0.shape
                     cv2.imwrite(save_path, im0)
                 else: 
                     if vid_path != save_path: 
@@ -153,7 +154,7 @@ def Custom_detect(img):
 
     return Image.fromarray(im0[:,:,::-1])
 inp = gr.Image(type="pil")
-output = gr.Image(type="numpy").style(full_width=True, height=10)
+output = gr.Image(type="pil")
 #gr.outputs.Image(type="filepath", label="Output")
 #.outputs.Textbox()  ,api_mode=True