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()