Spaces:
Runtime error
Runtime error
File size: 505 Bytes
d5da3d0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
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()
|