ritiksh commited on
Commit
cf8795e
·
1 Parent(s): c72a8a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ from PIL import Image
32
  if upload_file:
33
  test_sample = Image.open(upload_file)
34
 
35
- inputs = processor(images=image, return_tensors="pt").to(device)
36
  pixel_values = inputs.pixel_values.to(device)
37
 
38
  generated_ids = model.generate(pixel_values=pixel_values, max_length=50)
 
32
  if upload_file:
33
  test_sample = Image.open(upload_file)
34
 
35
+ inputs = processor(images=test_sample, return_tensors="pt").to(device)
36
  pixel_values = inputs.pixel_values.to(device)
37
 
38
  generated_ids = model.generate(pixel_values=pixel_values, max_length=50)