creaoy/AI_Long_Content_Generation

1from aitextgen import aitextgen2import gradio as gr3import os4from transformers import pipeline5from gradio import inputs6from gradio.inputs import Textbox7from gradio import outputs8 9 10ai=aitextgen(model='EleutherAI/gpt-neo-2.7B',to_gpu=False) # EleutherAI/gpt-neo-2.7B EleutherAI/gpt-neo-1.3B11# ai=aitextgen(model='EleutherAI/gpt-neo-1.3B',to_gpu=False)12 13def ai_text(Input):14  generated_text = ai.generate_one(max_length = 1000, prompt = Input, no_repeat_ngram_size = 3) #repetition_penalty = 1.9)15  #print(type(generated_text))16  return generated_text17 18 19title_ = "AI Long Content Generation"20description_ = "Converts short sentences into 1000 words"21output_text = gr.outputs.Textbox()22iface=gr.Interface(ai_text,"textbox", output_text, title=title_,description=description_)#.launch()23iface.launch()24 25 26 27#HF_TOKEN = os.environ.get("HF_TOKEN") 28#generator2 = gr.Interface.load("huggingface/EleutherAI/gpt-neo-2.7B", api_key=HF_TOKEN) # add api_key=HF_TOKEN to get over the quota error29#generator3 = gr.Interface.load("huggingface/EleutherAI/gpt-j-6B", api_key=HF_TOKEN)30#generator1 = gr.Interface.load("huggingface/gpt2-large", api_key=HF_TOKEN)31#gr.Parallel(generator1, generator2, generator3, inputs=gr.inputs.Textbox(lines=5, label="Enter a sentence to get another sentence."),32#            title=title, examples=examples).launch(share=False)