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

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="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
 
 
10
 
11
  # text2story
12
  def text2story(text):
13
+ story_generator = pipeline("text-generation", model="aspis/gpt2-genre-story-generation") # 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