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