ElvisTsang commited on
Commit
b545ce9
·
verified ·
1 Parent(s): acdb9cb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ def img2text(url):
9
  def text2story(text):
10
  story_generator = pipeline("text-generation", model="pranavpsv/gpt2-genre-story-generator", device_map="auto", return_full_text=False)
11
  prompt = f"A short story under 100 English words about: {text}."
12
- generated = story_generator(text, max_new_tokens=130, do_sample=True, temperature=0.7)
13
  story_text = generated[0]['generated_text']
14
  return story_text
15
 
 
9
  def text2story(text):
10
  story_generator = pipeline("text-generation", model="pranavpsv/gpt2-genre-story-generator", device_map="auto", return_full_text=False)
11
  prompt = f"A short story under 100 English words about: {text}."
12
+ generated = story_generator(text, max_new_tokens=140, do_sample=True, temperature=0.7)
13
  story_text = generated[0]['generated_text']
14
  return story_text
15