Spaces:
Runtime error
Runtime error
File size: 579 Bytes
6e47dae 9df383c 6e47dae |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
import gradio as gr
from aitextgen import aitextgen
def text_gen(text):
model_name = "EleutherAI/gpt-neo-125M"
model = aitextgen(model = model_name)
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()
|