deepBiz commited on
Commit
25c59d0
·
1 Parent(s): 98ec861

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ imgTransforms = transforms.Compose([
20
  transforms.ToTensor(),
21
  transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])])
22
 
23
- def predict(im):
24
  inp = Image.fromarray(inp.astype('uint8'), 'RGB')
25
  inp = imgTransforms(inp).unsqueeze(0)
26
  with torch.no_grad():
 
20
  transforms.ToTensor(),
21
  transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])])
22
 
23
+ def predict(inp):
24
  inp = Image.fromarray(inp.astype('uint8'), 'RGB')
25
  inp = imgTransforms(inp).unsqueeze(0)
26
  with torch.no_grad():