Update app.py

app.py

@@ -129,8 +129,7 @@ def Custom_detect(img):
                 cv2.waitKey(1)  
 
             if save_img:
-                if dataset.mode == 'image':
-                    im0.shape
+                if dataset.mode == 'image':                
                     cv2.imwrite(save_path, im0)
                 else: 
                     if vid_path != save_path: 
@@ -157,7 +156,7 @@ inp = gr.Image(type="pil")
 output = gr.Image(type="pil")
 #gr.outputs.Image(type="filepath", label="Output")
 #.outputs.Textbox()  ,api_mode=True
-css = ".output-image, .input-image {height: 40rem !important; width: 40% !important;}"
+css = ".output-image, .input-image {height: 40rem !important; width: 20rem !important;}"
 examples=[["Examples/Image1.jpg","Image1"],["Examples/Image2.jpg","Image2"],["Examples/Image29.jpg","Image3"]]
 
 io=gr.Interface(fn=Custom_detect, inputs=inp, outputs=output, title='Pot Hole Detection With Custom YOLOv7',examples=examples,cache_examples=False,css=css)