xujinheng666 commited on
Commit
ceb5c15
·
verified ·
1 Parent(s): a565167

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
- story_generator = pipeline("text-generation", model="mosaicml/mpt-7b-storywriter", trust_remote_code=True) # Corrected pipeline initialization
14
  story_text = story_generator(text, max_length=150, num_return_sequences=1) # Pass parameters here
15
  return story_text[0]["generated_text"] # Extract generated text
16
 
 
10
 
11
  # text2story
12
  def text2story(text):
13
+ story_generator = pipeline("text-generation", model="Monero/WizardLM-Uncensored-SuperCOT-StoryTelling-30b") # Corrected pipeline initialization
14
  story_text = story_generator(text, max_length=150, num_return_sequences=1) # Pass parameters here
15
  return story_text[0]["generated_text"] # Extract generated text
16