Spaces:

ancebuc
/

Grapes-segmentation

Runtime error

ancebuc commited on Apr 21, 2024

Commit

0970655

verified ·

1 Parent(s): 6d8dbab

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -61,7 +61,6 @@ class TargetMaskConvertTransform(ItemTransform):
 repo_id = "ancebuc/grapes-segmentation"
 learner = from_pretrained_fastai(repo_id)
-labels = learner.dls.vocab
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = torch.jit.load("unet.pth")

 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")