Soumen commited on
Commit
61ecb4f
·
1 Parent(s): 13a9962

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -105,7 +105,7 @@ def main():
105
  if st.button("Generate"):
106
  st.text("Using Hugging Face Trnsformer, Contrastive Search ..")
107
  output = model.generate(input_ids, max_length=128)
108
- st.success(tokenizer.decode(output[0], skip_special_tokens=True)))
109
  def change_photo_state():
110
  st.session_state["photo"]="done"
111
  st.subheader("Summary section, feed your image!")
 
105
  if st.button("Generate"):
106
  st.text("Using Hugging Face Trnsformer, Contrastive Search ..")
107
  output = model.generate(input_ids, max_length=128)
108
+ st.success(tokenizer.decode(output[0], skip_special_tokens=True))
109
  def change_photo_state():
110
  st.session_state["photo"]="done"
111
  st.subheader("Summary section, feed your image!")