Spaces:
Runtime error
Runtime error
| import gradio as gr | |
| from aitextgen import aitextgen | |
| ai = aitextgen(model="EleutherAI/gpt-neo-125M",to_gpu=True) | |
| def ai_text(inp): | |
| generated_text = ai.generate_one(max_length=1000, prompt = inp, no_repeat_ngram_size = 3) | |
| print(type(generated_text)) | |
| return generated_text | |
| output_text = gr.output.Textbox() | |
| gr.Interface(ai_text,'texbox',output_text,title="Ai Generated Blog using GPT-Neo", | |
| description="AI Generated Blog Content with GPT-Neo - via{aitextgen}").launch() | |