Update src/streamlit_app.py
Browse files- src/streamlit_app.py +1 -1
src/streamlit_app.py
CHANGED
|
@@ -12,7 +12,7 @@ def instantiate_gpt2(model_name: str,max_length_ : int, num_return_sequences : i
|
|
| 12 |
if model_name is "Qwen_v0.1":
|
| 13 |
results = pipe(
|
| 14 |
text,
|
| 15 |
-
max_new_tokens=max_length_
|
| 16 |
num_return_sequences=num_return_sequences,
|
| 17 |
do_sample=True,
|
| 18 |
top_p=0.95,
|
|
|
|
| 12 |
if model_name is "Qwen_v0.1":
|
| 13 |
results = pipe(
|
| 14 |
text,
|
| 15 |
+
max_new_tokens=max_length_,
|
| 16 |
num_return_sequences=num_return_sequences,
|
| 17 |
do_sample=True,
|
| 18 |
top_p=0.95,
|