Add files via upload

README.md

@@ -0,0 +1,13 @@
+---
+title: Fantasy Spell Generation with GPT2
+emoji: 🪄
+colorFrom: pink
+colorTo: blue
+sdk: gradio
+sdk_version: 3.16.2
+app_file: app.py
+pinned: true
+tags: ["text generation", "gpt", "spells", "rpg"]
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,58 @@
+import os
+import gradio as gr
+from transformers import pipeline
+
+HF_API = os.environ.get("HF_API")
+
+model = "aegrif/gpt2_spell_gen"
+tokenizer = "aegrif/gpt2_spell_gen"
+pipeline = pipeline(task="text-generation", model=model, tokenizer=tokenizer, use_auth_token=HF_API)
+
+
+def predict(text, temperature, top_k, top_p, max_length):
+    input_text = f"<|name|> {text} <|spell|>"
+    pipeline.model.config.pad_token_id = pipeline.model.config.eos_token_id
+    pipeline.model.config.temperature = temperature
+    pipeline.model.config.top_k = top_k
+    pipeline.model.config.top_p = top_p
+    predictions = pipeline(input_text, max_length=max_length, num_return_sequences=1)[0]["generated_text"]
+    spell_start = len(text) + 19
+    output = text + "\n\n" + predictions[spell_start:]
+    return output.strip()
+
+
+title = "# Spell generation with GPT-2"
+description = "## Generate your own spells"
+examples = [["Speak with Objects"], ["Summon Burley"], ["Moon Step"], ["Burden of the Gods"], ["Shape Rock"], ["Bard's Laughter"], ["Mundane Foresight"], ["Word of Cancellation"]]
+
+with gr.Blocks(css="#spell-row {justify-content: flex-start; }") as interface:
+    gr.Markdown(title)
+    gr.Markdown(description)
+    with gr.Row(variant="compact", elem_id="spell-row"):
+        with gr.Column(scale=1):
+            name = gr.Textbox(lines=1, label="Spell Name", placeholder="Enter your spell name here...")
+    with gr.Row(variant="compact", elem_id="spell-row"):
+        with gr.Column(scale=3):
+            gr.Examples(examples, inputs=name)
+    with gr.Row(variant="compact", elem_id="spell-row"):
+        gr.Markdown("## Model Settings")
+    with gr.Row(variant="compact", elem_id="spell-row"):
+        with gr.Column(scale=1):
+            max_length = gr.Slider(minimum=50, maximum=800, step=50, value=400, label="Max Length")
+        with gr.Column(scale=1):
+            temperature = gr.Slider(minimum=0.1, maximum=float(1.9), step=0.1, value=float(1), label="Temperature")
+    with gr.Row(variant="compact", elem_id="spell-row"):
+        with gr.Column(scale=1):
+            top_k = gr.Slider(minimum=0, maximum=1000, step=10, value=50, label="Top K")
+        with gr.Column(scale=1):
+            top_p = gr.Slider(minimum=0.0, maximum=1.0, step=0.1, value=1, label="Top P")
+    with gr.Row(variant="compact", elem_id="spell-row"):
+        output = gr.Textbox(label="Generated Spell", placeholder="Your spell will appear here.")
+    generate_btn = gr.Button("Generate Spell")
+    with gr.Row(variant="compact", elem_id="spell-row"):
+        gr.Markdown(
+            "**Max Length**: The maximum length of the generated spell.\n\n**Temperature**: The randomness of the generated spell. Higher values are more random, lower values are more deterministic.\n\n**Top K**: The number of highest probability vocabulary tokens to keep for top-k-filtering.\n\n**Top P**: The cumulative probability for top-p-filtering.")
+
+    generate_btn.click(fn=predict, inputs=[name,temperature,top_k,top_p, max_length], outputs=output)
+
+interface.launch()

gitattributes.txt

@@ -0,0 +1,34 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

requirements.txt

@@ -0,0 +1,4 @@
+gradio
+transformers
+huggingface_hub
+torch