Ignaciobfp commited on
Commit
533d4eb
·
1 Parent(s): 3610887

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -4,11 +4,12 @@ from fastai.vision.all import *
4
  import PIL
5
  import torchvision.transforms as transforms
6
 
7
- repo_id = "Ignaciobfp/segmentacion-dron-marras"
8
- learner = from_pretrained_fastai(repo_id)
9
 
10
  device = torch.device("cpu")
11
- model = learner.model
 
12
  model = model.cpu()
13
 
14
  def transform_image(image):
 
4
  import PIL
5
  import torchvision.transforms as transforms
6
 
7
+ #repo_id = "Ignaciobfp/segmentacion-dron-marras"
8
+ #learner = from_pretrained_fastai(repo_id)
9
 
10
  device = torch.device("cpu")
11
+ #model = learner.model
12
+ model = torch.jit.load("modelo_marras.pth")
13
  model = model.cpu()
14
 
15
  def transform_image(image):