Spaces:

azaninello
/

gpt2-general

Runtime error

azaninello commited on Jun 27, 2022

Commit

81d2dea

•

1 Parent(s): 6cf0c2b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ shroom_generator = pipeline("text-generation", model=AutoModelWithLMHead.from_pr
     max_length=100,
     top_k=50,
     top_p=0.95,
-    repetition_penalty=0.1)
 def generator(inizia_la_storia = ''):
   shroom_result = shroom_generator(inizia_la_storia, max_length=520)

     max_length=100,
     top_k=50,
     top_p=0.95,
+    repetition_penalty=9.5)
 def generator(inizia_la_storia = ''):
   shroom_result = shroom_generator(inizia_la_storia, max_length=520)