Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -22,18 +22,22 @@ import torch.nn.functional as F
|
|
22 |
|
23 |
|
24 |
# # repo_id = "YOUR_USERNAME/YOUR_LEARNER_NAME"
|
25 |
-
repo_id = "islasher/segm-grapes"
|
26 |
-
repo_id='islasher/segm-grapes'
|
27 |
|
28 |
|
29 |
# # Definimos una función que se encarga de llevar a cabo las predicciones
|
30 |
|
31 |
-
from fastai.learner import load_learner
|
32 |
|
33 |
# # Cargar el modelo y el tokenizador
|
34 |
-
learn = load_learner(repo_id)
|
35 |
#learner = from_pretrained_fastai(repo_id)
|
36 |
|
|
|
|
|
|
|
|
|
37 |
import torchvision.transforms as transforms
|
38 |
def transform_image(image):
|
39 |
my_transforms = transforms.Compose([transforms.ToTensor(),
|
|
|
22 |
|
23 |
|
24 |
# # repo_id = "YOUR_USERNAME/YOUR_LEARNER_NAME"
|
25 |
+
# repo_id = "islasher/segm-grapes"
|
26 |
+
# repo_id='islasher/segm-grapes'
|
27 |
|
28 |
|
29 |
# # Definimos una función que se encarga de llevar a cabo las predicciones
|
30 |
|
31 |
+
# from fastai.learner import load_learner
|
32 |
|
33 |
# # Cargar el modelo y el tokenizador
|
34 |
+
# learn = load_learner(repo_id)
|
35 |
#learner = from_pretrained_fastai(repo_id)
|
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 |
def transform_image(image):
|
43 |
my_transforms = transforms.Compose([transforms.ToTensor(),
|