Update app.py

app.py

@@ -25,8 +25,7 @@ def yolov7_inference(
     Returns:
         Rendered image
     """
-
-    model = torch.hub.load('kadirnar/yolov7-v0.1', 'custom', path='skin_burn.pt', source='local', hf_model=True, device="cpu")
+    model = yolov7.load(model_path, device="cpu", hf_model=True, trace=False)
     model.conf = conf_threshold
     model.iou = iou_threshold
     results = model([image], size=image_size)
@@ -37,7 +36,7 @@ inputs = [
     gr.inputs.Image(type="pil", label="Input Image"),
     gr.inputs.Dropdown(
         choices=[
-            "skin_burn",
+            "skin_burn.pt",
             "Other"
         ],
         default="skin_burn",