upload huggingface spaces

app.py

@@ -0,0 +1,21 @@
+import gradio as gr
+from aitextgen import aitextgen
+
+
+def text_gen(text):
+
+    model_name = "EleutherAI/gpt-neo-125M"
+    model = aitextgen(model = model_name, to_gpu = True)
+    text_generated = model.generate_one(max_length=1000, prompt = text, no_repeat_ngram_size = 3)
+
+    return text_generated
+
+
+output_text = gr.outputs.Textbox()
+
+gr.Interface(text_gen,
+             inputs = gr.inputs.Textbox(label = "Input Text"),
+             outputs = gr.outputs.Textbox(),
+             title = "Blog Generator using AI",
+             description = "Create a 1000 word blog on any topic").launch()
+

requirements.txt

@@ -0,0 +1,2 @@
+aitextgen
+gradio