Spaces:
Runtime error
Runtime error
Commit ·
4f7c840
1
Parent(s): 51f9043
added space code
Browse files
app.py
ADDED
|
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import gradio as gr
|
| 2 |
+
from peft import PeftModel, PeftConfig
|
| 3 |
+
from transformers import AutoModelForCausalLM
|
| 4 |
+
|
| 5 |
+
config = PeftConfig.from_pretrained("Electricarchmage/ApocryphaGenerator")
|
| 6 |
+
model = AutoModelForCausalLM.from_pretrained("EleutherAI/gpt-neox-20b")
|
| 7 |
+
model = PeftModel.from_pretrained(model, "Electricarchmage/ApocryphaGenerator")
|
| 8 |
+
|
| 9 |
+
apocrypha_gen = gr.Interface.load("Electricarchmage/ApocryphaGenerator")
|
| 10 |
+
description = "Generate Apocrypha (just fyi this thing pulls up all 20 billion parameters in the model as it executes here, so you might be here for a while)"
|
| 11 |
+
title = "Apocrypha Generator using GPT-NeoX-20b"
|
| 12 |
+
examples[["Some apocrypha about Noah's Ark: "]]
|
| 13 |
+
|
| 14 |
+
Series(description = description, title = title, inputs = gr.inputs.Textbox(lines = 10)).launch()
|