Spaces:

ancebuc
/

Grapes-segmentation

Runtime error

ancebuc commited on Apr 21, 2024

Commit

7213c65

verified ·

1 Parent(s): 0970655

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -63,7 +63,6 @@ repo_id = "ancebuc/grapes-segmentation"
 learner = from_pretrained_fastai(repo_id)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model = torch.jit.load("unet.pth")
 model = model.cpu()
 model.eval()

 learner = from_pretrained_fastai(repo_id)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = model.cpu()
 model.eval()