Update app.py

app.py

@@ -4,11 +4,16 @@ from fastai.vision.all import *
 
 
 # repo_id = "YOUR_USERNAME/YOUR_LEARNER_NAME"
-repo_id = "igmarco/grapes-semanticsegmentation"
+# repo_id = "igmarco/grapes-semanticsegmentation"
 
-learner = from_pretrained_fastai(repo_id)
+# learner = from_pretrained_fastai(repo_id)
 
 import torchvision.transforms as transforms
+import PIL
+
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu") 
+model = torch.jit.load("Pr1.pth")
+model = model.cpu()
 
 def transform_image(image):
     my_transforms = transforms.Compose([transforms.ToTensor(),