Spaces:
Runtime error
Runtime error
Adding transformers library to import pipeline
Browse files
app.py
CHANGED
@@ -1,4 +1,5 @@
|
|
1 |
import gradio as gr
|
|
|
2 |
|
3 |
def generate_story(story):
|
4 |
story_gen = pipeline("text-generation", "pranavpsv/gpt2-genre-story-generator")
|
|
|
1 |
import gradio as gr
|
2 |
+
from transformers import pipeline
|
3 |
|
4 |
def generate_story(story):
|
5 |
story_gen = pipeline("text-generation", "pranavpsv/gpt2-genre-story-generator")
|