namuisam commited on
Commit
bfa949b
·
verified ·
1 Parent(s): 86b1ea3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,14 +9,14 @@ def img2text(url):
9
  return text
10
 
11
  def text2story(text):
12
- text_generation_model = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2")
13
  story_text = text_generation_model(
14
  text,
15
  min_length=50,
16
  max_length=100,
17
  do_sample=True,
18
  early_stopping=True,
19
- top_p=0.4
20
  )[0]["generated_text"]
21
  return story_text
22
 
 
9
  return text
10
 
11
  def text2story(text):
12
+ text_generation_model = pipeline("text-generation", model="aspis/gpt2-genre-story-generation")
13
  story_text = text_generation_model(
14
  text,
15
  min_length=50,
16
  max_length=100,
17
  do_sample=True,
18
  early_stopping=True,
19
+ top_p=0.6
20
  )[0]["generated_text"]
21
  return story_text
22