Spaces:

sanjanatule
/

ERA_Assignment13

Sleeping

sanjanatule commited on Aug 25, 2023

Commit

49b5807

1 Parent(s): 9e28e3e

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_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_improved_model.ckpt")
       # bboxes, gradcam
       anchors  = (torch.tensor(config.ANCHORS) * torch.tensor(config.S).unsqueeze(1).unsqueeze(1).repeat(1, 3, 2))