Update app.py

app.py

@@ -22,18 +22,22 @@ import torch.nn.functional as F
 
 
 # # repo_id = "YOUR_USERNAME/YOUR_LEARNER_NAME"
-repo_id = "islasher/segm-grapes"
-repo_id='islasher/segm-grapes'
+# repo_id = "islasher/segm-grapes"
+# repo_id='islasher/segm-grapes'
 
 
 # # Definimos una función que se encarga de llevar a cabo las predicciones
 
-from fastai.learner import load_learner
+# from fastai.learner import load_learner
 
 # # Cargar el modelo y el tokenizador
-learn = load_learner(repo_id)
+# learn = load_learner(repo_id)
 #learner = from_pretrained_fastai(repo_id)
 
+from huggingface_hub import from_pretrained_fastai
+
+learn = from_pretrained_fastai("islasher/segm-grapes")
+
 import torchvision.transforms as transforms
 def transform_image(image):
     my_transforms = transforms.Compose([transforms.ToTensor(),