PyaeSoneK commited on
Commit
cb5e41b
·
1 Parent(s): a92c384

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -36,7 +36,7 @@ pipe = pipeline("text-generation",
36
  max_new_tokens = 512,
37
  do_sample=True,
38
  top_k=30,
39
- num_return_sequences=1,
40
  eos_token_id=tokenizer.eos_token_id
41
  )
42
 
@@ -114,7 +114,7 @@ text = st.text_input('Plug in your prompt here')
114
 
115
  if text:
116
  response = llm_chain.run(text)
117
- st.write(response)
118
 
119
 
120
 
 
36
  max_new_tokens = 512,
37
  do_sample=True,
38
  top_k=30,
39
+ num_return_sequences=4,
40
  eos_token_id=tokenizer.eos_token_id
41
  )
42
 
 
114
 
115
  if text:
116
  response = llm_chain.run(text)
117
+ st.write(parse_text(response))
118
 
119
 
120