ErikDaska commited on
Commit
30cce29
·
verified ·
1 Parent(s): 461a5c5

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -8,7 +8,7 @@ token = os.environ.get("token")
8
 
9
 
10
  def instantiate_gpt2(max_length_ : int, num_return_sequences : int, text : str) -> dict:
11
- pipe = pipeline(task='text-generation', model='Iscte-Sintra/GPT2-Kriolu', tokenizer='Iscte-Sintra/GPT2-Kriolu', token=token)
12
  results = pipe(text, max_length=max_length_, num_return_sequences=num_return_sequences)
13
  return results
14
 
 
8
 
9
 
10
  def instantiate_gpt2(max_length_ : int, num_return_sequences : int, text : str) -> dict:
11
+ pipe = pipeline(task='text-generation', model='Iscte-Sintra/GPT2-Kriolu', tokenizer='Iscte-Sintra/GPT2-Kriolu', token=token, truncation=True)
12
  results = pipe(text, max_length=max_length_, num_return_sequences=num_return_sequences)
13
  return results
14