Soumen commited on
Commit
bb28fb1
·
1 Parent(s): 9531d63

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -105,9 +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("Output:\n" + 100 * '-')
109
  st.success(tokenizer.decode(output[0], skip_special_tokens=True)))
110
- st.success("" + 100 * '-')
111
  def change_photo_state():
112
  st.session_state["photo"]="done"
113
  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!")