ErikDaska commited on
Commit
a86516e
·
verified ·
1 Parent(s): 5197a07

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. 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,