fyyb commited on
Commit
b432533
·
verified ·
1 Parent(s): 28db8dd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ def img2text(url):
11
 
12
  # text2story
13
  def text2story(text):
14
- text_generation_model = pipeline("text-generation", model="microsoft/Phi-4-mini-instruct", trust_remote_code=True)
15
  story_text = text_generation_model(text, min_length=100, max_length=150)[0]["generated_text"]
16
  return story_text
17
 
 
11
 
12
  # text2story
13
  def text2story(text):
14
+ text_generation_model = pipeline("text-generation", model="pranavpsv/gpt2-genre-story-generator", trust_remote_code=True)
15
  story_text = text_generation_model(text, min_length=100, max_length=150)[0]["generated_text"]
16
  return story_text
17