Sarah Yakum commited on
Commit
84e17c7
1 Parent(s): 11aec48
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ model.to(device)
17
 
18
 
19
  if file_name is not None:
20
- image = Image.open(img_name)
21
  image.show()
22
 
23
  pixel_values = feature_extractor(image, return_tensors="pt").pixel_values.to(device)
 
17
 
18
 
19
  if file_name is not None:
20
+ image = Image.open(file_name)
21
  image.show()
22
 
23
  pixel_values = feature_extractor(image, return_tensors="pt").pixel_values.to(device)