Update app.py

app.py

@@ -20,7 +20,6 @@ os.system('git clone https://github.com/WongKinYiu/yolov7')
 
  
 def Custom_detect(img):
-    #if model =='Yolo_v7_Custom_model':
     model='best' 
     parser = argparse.ArgumentParser()
     parser.add_argument('--weights', nargs='+', type=str, default=model+".pt", help='model.pt path(s)')
@@ -154,7 +153,7 @@ def Custom_detect(img):
 
     return Image.fromarray(im0[:,:,::-1])
 inp = gr.Image(type="pil")
-output = gr.Image(type="pil").style(full_width=True, height=60)
+output = gr.Image(type="pil").style(full_width=True, height=10)
 #gr.outputs.Image(type="filepath", label="Output")
 #.outputs.Textbox()  ,api_mode=True