mirari commited on
Commit
d0b6c0e
1 Parent(s): 902fb02

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ model = model.cpu()
21
  model.eval()
22
 
23
  def transform_image(image):
24
- my_transforms = transforms.Compose([transforms.ToTensor(), transforms.Normalzie([0.485, 0.456, 0.406],[0.229, 0.224, 0.225])])
25
  return my_transforms(image).unsqueeze(0).to(device)
26
 
27
  def predict(img):
 
21
  model.eval()
22
 
23
  def transform_image(image):
24
+ my_transforms = transforms.Compose([transforms.ToTensor(), transforms.Normalize([0.485, 0.456, 0.406],[0.229, 0.224, 0.225])])
25
  return my_transforms(image).unsqueeze(0).to(device)
26
 
27
  def predict(img):