Update app.py
Browse files
app.py
CHANGED
|
@@ -13,9 +13,6 @@ def process_image(image, prompt):
|
|
| 13 |
# Process the image and prompt using the processor
|
| 14 |
inputs = processor(image.convert("RGB"), prompt, return_tensors="pt")
|
| 15 |
|
| 16 |
-
# Print the inputs to debug
|
| 17 |
-
print("Processor outputs:", inputs)
|
| 18 |
-
|
| 19 |
try:
|
| 20 |
# Generate output from the model
|
| 21 |
output = model.generate(**inputs, max_new_tokens=20)
|
|
|
|
| 13 |
# Process the image and prompt using the processor
|
| 14 |
inputs = processor(image.convert("RGB"), prompt, return_tensors="pt")
|
| 15 |
|
|
|
|
|
|
|
|
|
|
| 16 |
try:
|
| 17 |
# Generate output from the model
|
| 18 |
output = model.generate(**inputs, max_new_tokens=20)
|