Leo Liu commited on
Commit
787878d
·
verified ·
1 Parent(s): 92eb71d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ def img2text(url):
15
 
16
  # text2story
17
  def text2story(text):
18
- pipe = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2",max_new_tokens=300, min_new_tokens=100)
19
  story_text = pipe(text)[0]['generated_text']
20
  return story_text
21
 
 
15
 
16
  # text2story
17
  def text2story(text):
18
+ pipe = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2",max_new_tokens=150, min_new_tokens=130)
19
  story_text = pipe(text)[0]['generated_text']
20
  return story_text
21