Spaces:
Build error
Build error
Update app.py
Browse files
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=
|
| 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)
|