Centauri_Pilot / app.py
ParthRangarajan's picture
version maintainance starts
dff0197
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=False)
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",
examples= [
['Chocolate gives bad skin'],
['India has the highest population'],
['The moon is not a planet'],
['Who is Alexander the Great?']],
theme='dark-peach',
description="V1 of Generating Blog Content using GPT-Neo by implementing aitextgen and Gradio").launch(inline=False)