Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -19,11 +19,12 @@ pipeline=transformers.pipeline(
|
|
| 19 |
max_length=64,
|
| 20 |
do_sample=True,
|
| 21 |
top_k=10,
|
|
|
|
| 22 |
num_return_sequences=1,
|
| 23 |
eos_token_id=tokenizer.eos_token_id
|
| 24 |
)
|
| 25 |
|
| 26 |
-
llm=HuggingFacePipeline(pipeline=pipeline, model_kwargs={'temperature':0})
|
| 27 |
|
| 28 |
def greet(prompt):
|
| 29 |
|
|
|
|
| 19 |
max_length=64,
|
| 20 |
do_sample=True,
|
| 21 |
top_k=10,
|
| 22 |
+
top_p=5,
|
| 23 |
num_return_sequences=1,
|
| 24 |
eos_token_id=tokenizer.eos_token_id
|
| 25 |
)
|
| 26 |
|
| 27 |
+
llm=HuggingFacePipeline(pipeline=pipeline, model_kwargs={'temperature':0.1})
|
| 28 |
|
| 29 |
def greet(prompt):
|
| 30 |
|