Update app.py

app.py

@@ -10,14 +10,14 @@ FILENAME = "skin_burn.pt"
 yolov7_custom_weights = hf_hub_download(repo_id=REPO_ID, filename=FILENAME,repo_type='space')
 
 model = torch.hub.load('WongKinYiu/yolov7:main',model='custom', path_or_model=yolov7_custom_weights, force_reload=True)
-def object_detection(im, size=416):
+def object_detection(im, size=614):
     results = model(im) 
     results.render()  
     return Image.fromarray(results.imgs[0])
 
 title = "Yolov7 Skin Burn Detection"
 
-image = gr.inputs.Image(shape=(416,416), image_mode="RGB", source="upload", label="Upload Image", optional=False)
+image = gr.inputs.Image(shape=(614,614), image_mode="RGB", source="upload", label="Upload Image", optional=False)
 outputs = gr.outputs.Image(type="pil", label="Output Image")