Create app.py

app.py

@@ -0,0 +1,62 @@
+Hugging Face's logo
+Hugging Face
+Search models, datasets, users...
+Models
+Datasets
+Spaces
+Docs
+Solutions
+Pricing
+
+
+
+Spaces:
+
+mshook
+/
+gpt2-story-generator-example Copied
+private
+
+Logs
+App
+Files
+Community
+Settings
+gpt2-story-generator-example
+/
+app.py
+mshook's picture
+mshook
+Update app.py
+26c39e6
+43 minutes ago
+raw
+history
+blame
+edit
+delete
+735 Bytes
+import gradio as gr
+from transformers import pipeline
+
+#generator = pipeline('text-generation', model='gpt2-xl')
+generator = pipeline('text-generation', model='pranavpsv/gpt2-genre-story-generator')
+
+def generate(text):
+    result = generator(text, max_length=100, num_return_sequences=1)
+    return result[0]["generated_text"]
+
+examples = [
+    ["<BOS> <sci_fi> After discovering time travel,"],
+    ["<BOS> <superhero> Batman"],
+    ["<BOS> <drama> Legolas and Gimli advanced on the orcs, raising their weapons with a harrowing war cry"],
+]
+
+demo = gr.Interface(
+    fn=generate,
+    inputs=gr.inputs.Textbox(lines=5, label="Input Text"),
+    outputs=gr.outputs.Textbox(label="Generated Text"),
+    examples=examples
+)
+
+demo.launch()