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