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)
|