vicgalle commited on
Commit
d41e1d1
1 Parent(s): f920d3a
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -63,10 +63,11 @@ if uploaded_image is not None:
63
  with st.spinner('Generating caption...'):
64
  output = model.generate(**inputs, max_length=1000)
65
 
 
66
  out = processor.decode(output[0], skip_special_tokens=True)[len(prompt) :]
67
 
68
  # Display the extracted text
69
- st.text_area("Coffe machine caption", out, height=300)
70
 
71
 
72
 
 
63
  with st.spinner('Generating caption...'):
64
  output = model.generate(**inputs, max_length=1000)
65
 
66
+ print(output)
67
  out = processor.decode(output[0], skip_special_tokens=True)[len(prompt) :]
68
 
69
  # Display the extracted text
70
+ st.text_area("Coffe machine caption", out, height=100)
71
 
72
 
73