Nehal Gajraj commited on
Commit ·
6cb6a1d
1
Parent(s): c759073
app.py
CHANGED
|
@@ -31,7 +31,7 @@ if user_input:
|
|
| 31 |
]
|
| 32 |
|
| 33 |
# Generate response
|
| 34 |
-
response = pipe(messages)
|
| 35 |
|
| 36 |
# Display the result
|
| 37 |
st.write("Model response:")
|
|
|
|
| 31 |
]
|
| 32 |
|
| 33 |
# Generate response
|
| 34 |
+
response = pipe(messages, max_new_tokens=512, do_sample=True, temperature=0.7)
|
| 35 |
|
| 36 |
# Display the result
|
| 37 |
st.write("Model response:")
|