Update app.py
Browse files
app.py
CHANGED
@@ -15,7 +15,7 @@ def launch(input):
|
|
15 |
for file in input:
|
16 |
inputs.append(processor(Image.open(file.name), return_tensors="pt"))
|
17 |
|
18 |
-
out = model.generate(
|
19 |
return processor.decode(out[0], skip_special_tokens=True)
|
20 |
|
21 |
description = "Simple BLIP test app for image captioning."
|
|
|
15 |
for file in input:
|
16 |
inputs.append(processor(Image.open(file.name), return_tensors="pt"))
|
17 |
|
18 |
+
out = model.generate(*inputs)
|
19 |
return processor.decode(out[0], skip_special_tokens=True)
|
20 |
|
21 |
description = "Simple BLIP test app for image captioning."
|