micrem73 commited on
Commit
8d2d3cb
·
1 Parent(s): 2f9ac1b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ shroom_generator = pipeline("text-generation", model=AutoModelWithLMHead.from_pr
17
  repetition_penalty=9.5 )
18
 
19
  def generator(inizia_la_storia = ''):
20
- shroom_result = shroom_generator(inizia_la_storia, max_length=520)
21
  return shroom_result[0]["generated_text"]
22
 
23
 
 
17
  repetition_penalty=9.5 )
18
 
19
  def generator(inizia_la_storia = ''):
20
+ shroom_result = shroom_generator(inizia_la_storia, max_length=100)
21
  return shroom_result[0]["generated_text"]
22
 
23