Predecessor to E2E Centauri Text Gen Web App

app.py

@@ -0,0 +1,13 @@
+import gradio as gr
+from aitextgen import aitextgen
+# Downloading gpt neo from hugging face model hub
+ai= aitextgen(model='EleutherAI/gpt-neo-125M', to_gpu=True)
+def ai_text(input):
+    # returning text as string for gradio
+    generated_text= ai.generate_one(max_length=1000, prompt=input, no_repeat_ngram_size=3)
+    print(generated_text)
+    return generated_text
+ output_text=gr.outputs.Textbox()
+gr.Interface(ai_text, "textbox", 
+             output_text, title="Centauri Pilot", 
+             description="Generating Blog Content using GPT-Neo by implementing aitextgen and Gradio").launch(inline=False)