Spaces:
Runtime error
Runtime error
Update app.py
Browse filesadd test print
app.py
CHANGED
|
@@ -35,6 +35,7 @@ def inference(raw_image, model_n, strategy):
|
|
| 35 |
captions = model.generate(**input, generation_config=config)
|
| 36 |
caption = processor.decode(captions[0], skip_special_tokens=True)
|
| 37 |
caption = caption.replace(' ', '')
|
|
|
|
| 38 |
return 'caption: '+caption
|
| 39 |
|
| 40 |
|
|
|
|
| 35 |
captions = model.generate(**input, generation_config=config)
|
| 36 |
caption = processor.decode(captions[0], skip_special_tokens=True)
|
| 37 |
caption = caption.replace(' ', '')
|
| 38 |
+
print(caption)
|
| 39 |
return 'caption: '+caption
|
| 40 |
|
| 41 |
|