Soumen commited on
Commit
adb2b76
·
1 Parent(s): e689ea0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -104,8 +104,8 @@ def main():
104
  if st.button("Generate"):
105
  tokenizer, model = load_models()
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!")
 
104
  if st.button("Generate"):
105
  tokenizer, model = load_models()
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!")