Update app.py
Browse files
app.py
CHANGED
@@ -77,7 +77,7 @@ def generate_caption_and_image(image, f, p, d):
|
|
77 |
|
78 |
|
79 |
|
80 |
-
caption2 =
|
81 |
|
82 |
|
83 |
inputs = processor(image, return_tensors="pt", padding=True, truncation=True, max_length=250)
|
|
|
77 |
|
78 |
|
79 |
|
80 |
+
caption2 = processor1.decode(out[0], skip_special_tokens=True)
|
81 |
|
82 |
|
83 |
inputs = processor(image, return_tensors="pt", padding=True, truncation=True, max_length=250)
|