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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -103,7 +103,7 @@ def main():
103
  input_ids = tokenizer(message, return_tensors='pt').input_ids
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():
 
103
  input_ids = tokenizer(message, return_tensors='pt').input_ids
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():