jerilseb commited on
Commit
3f76519
1 Parent(s): 3299692

update stuff

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ transform = transforms.Compose([
41
  def predict(image):
42
  image = image['composite']
43
  image = Image.fromarray(image)
44
- input = transform(image).unsqeeze(0)
45
  print(image.shape)
46
 
47
  with torch.no_grad():
 
41
  def predict(image):
42
  image = image['composite']
43
  image = Image.fromarray(image)
44
+ input = transform(image).unsqueeze(0)
45
  print(image.shape)
46
 
47
  with torch.no_grad():