islasher commited on
Commit
66cbf63
1 Parent(s): 6786150

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -36,7 +36,7 @@ import torch.nn.functional as F
36
 
37
  from huggingface_hub import from_pretrained_fastai
38
 
39
- learn = from_pretrained_fastai("islasher/segm-grapes")
40
 
41
  import torchvision.transforms as transforms
42
 
@@ -70,6 +70,9 @@ class TargetMaskConvertTransform(ItemTransform):
70
 
71
 
72
 
 
 
 
73
  def transform_image(image):
74
  my_transforms = transforms.Compose([transforms.ToTensor(),
75
  transforms.Normalize(
 
36
 
37
  from huggingface_hub import from_pretrained_fastai
38
 
39
+
40
 
41
  import torchvision.transforms as transforms
42
 
 
70
 
71
 
72
 
73
+
74
+ learn = from_pretrained_fastai("islasher/segm-grapes")
75
+
76
  def transform_image(image):
77
  my_transforms = transforms.Compose([transforms.ToTensor(),
78
  transforms.Normalize(