Jessiesj commited on
Commit
86d4515
·
verified ·
1 Parent(s): 1eab4a7

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="mosaicml/mpt-7b-storywriter", trust_remote_code=True)
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("text-generation", model="TheBloke/storytime-13B-GPTQ")
14
  # story_text = "" # to be completed
15
  story_text = text_to_story(text)[0]["generated_text"]
16
  return story_text