Spaces:

sanjanatule
/

ERA_Assignment13

Sleeping

sanjanatule commited on Aug 22, 2023

Commit

5e22169

1 Parent(s): 02723e8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -84,7 +84,7 @@ with gr.Blocks() as demo:
       # load model
       yololit = litmodelclass.LitYolo()
-      inference_model = yololit.load_from_checkpoint("yolo3_last_model.ckpt")
       # bboxes, gradcam
       anchors  = (torch.tensor(config.ANCHORS) * torch.tensor(config.S).unsqueeze(1).unsqueeze(1).repeat(1, 3, 2))

       # load model
       yololit = litmodelclass.LitYolo()
+      inference_model = yololit.load_from_checkpoint("yolo3_model.ckpt")
       # bboxes, gradcam
       anchors  = (torch.tensor(config.ANCHORS) * torch.tensor(config.S).unsqueeze(1).unsqueeze(1).repeat(1, 3, 2))