Jessiesj commited on
Commit
c0a50ad
·
verified ·
1 Parent(s): 9a7f42f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ def img2text(url):
10
 
11
  # text2story
12
  def text2story(text):
13
- text_to_story = pipeline("text-generation", model="isarth/distill_gpt2_story_generator")
14
  # story_text = "" # to be completed
15
  story_text = text_to_story(text)[0]["generated_text"]
16
  return story_text
 
10
 
11
  # text2story
12
  def text2story(text):
13
+ text_to_story = pipeline("text2text-generation", model="Salesforce/codet5p-220m-py")
14
  # story_text = "" # to be completed
15
  story_text = text_to_story(text)[0]["generated_text"]
16
  return story_text