Update app.py

app.py

@@ -62,6 +62,21 @@ repo_id = "ancebuc/grapes-segmentation"
 
 learner = from_pretrained_fastai(repo_id)
 
+aux = learner.model
+aux = aux.cpu()
+
+import torchvision.transforms as transforms
+img = PILImage.create(path_images/'train/color_206.jpg')
+transformer=transforms.Compose([transforms.Resize((480,640)),
+                                transforms.ToTensor(),
+                                transforms.Normalize(
+                                                    [0.485, 0.456, 0.406],
+                                                    [0.229, 0.224, 0.225])])
+img=transformer(img).unsqueeze(0)
+img=img.cpu()
+
+traced_cell=torch.jit.trace(aux, (img))
+
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = model.cpu()
 model.eval()