Add application file
Browse files
app.py
CHANGED
|
@@ -15,7 +15,7 @@ shroom_generator = pipeline("text-generation", model=AutoModelWithLMHead.from_pr
|
|
| 15 |
top_k=50,
|
| 16 |
top_p=0.95,
|
| 17 |
repetition_penalty=9.5,
|
| 18 |
-
temperature=
|
| 19 |
|
| 20 |
def generator(inizia_la_storia = ''):
|
| 21 |
shroom_result = shroom_generator(inizia_la_storia, max_length=520)
|
|
|
|
| 15 |
top_k=50,
|
| 16 |
top_p=0.95,
|
| 17 |
repetition_penalty=9.5,
|
| 18 |
+
temperature=1.1 )
|
| 19 |
|
| 20 |
def generator(inizia_la_storia = ''):
|
| 21 |
shroom_result = shroom_generator(inizia_la_storia, max_length=520)
|