paugar commited on
Commit
0d83e38
·
verified ·
1 Parent(s): 8166e48

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ def load_img (filename):
21
  return img
22
 
23
  def process_img(image):
24
- y = img.to(device)
25
 
26
  with torch.no_grad():
27
  result = model(y)
 
21
  return img
22
 
23
  def process_img(image):
24
+ y = image.to(device)
25
 
26
  with torch.no_grad():
27
  result = model(y)