SmilingTree commited on
Commit
266b51d
·
verified ·
1 Parent(s): 9adffde

return to gpt2_story generator

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,8 +1,8 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- # pipeline = pipeline("text-generation", model="isarth/distill_gpt2_story_generator")
5
- pipeline = pipeline("text-generation", model="tohur/natsumura-storytelling-rp-1.0-llama-3.1-8b")
6
 
7
  def text_generate(input_text):
8
  result = pipeline(input_text)
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ pipeline = pipeline("text-generation", model="isarth/distill_gpt2_story_generator")
5
+ # pipeline = pipeline("text-generation", model="tohur/natsumura-storytelling-rp-1.0-llama-3.1-8b")
6
 
7
  def text_generate(input_text):
8
  result = pipeline(input_text)