Update src/streamlit_app.py
Browse files- 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 |
|