Update app.py
Browse files
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=
|
| 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 |
|