inigo99 commited on
Commit
8477d2b
·
1 Parent(s): 826805f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,11 +7,11 @@ from fastai.vision.all import *
7
  from fastai.metrics import *
8
  from fastai.data.all import *
9
  from fastai.callback import *
 
10
 
11
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
12
  model = torch.jit.load("hrnet.pth")
13
 
14
- import torchvision.transforms as transforms
15
  def transform_image(image):
16
  my_transforms = transforms.Compose([transforms.ToTensor(),
17
  transforms.Normalize(
 
7
  from fastai.metrics import *
8
  from fastai.data.all import *
9
  from fastai.callback import *
10
+ import torchvision.transforms as transforms
11
 
12
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
13
  model = torch.jit.load("hrnet.pth")
14
 
 
15
  def transform_image(image):
16
  my_transforms = transforms.Compose([transforms.ToTensor(),
17
  transforms.Normalize(