Update app.py

app.py

@@ -1,20 +1,17 @@
+from aitextgen import aitextgen
 import gradio as gr
 
-title = "GPT-J-6B"
 
-description = "Gradio Demo for GPT-J 6B, a transformer model trained \
-using Ben Wang's Mesh Transformer JAX. 'GPT-J' refers to the class of \
-model, while '6B' represents the number of trainable parameters."
+ai=aitextgen(model='EleutherAI/gpt-neo-1.3B',to_gpu=False) # EleutherAI/gpt-neo-2.7B EleutherAI/gpt-neo-1.3B
 
+def ai_text(Input):
+  generated_text = ai.generate_one(max_length = 500, prompt = Input, no_repeat_ngram_size = 3) #repetition_penalty = 1.9)
+  #print(type(generated_text))
+  return generated_text
 
-examples = [
-    ['The tower is 324 metres (1,063 ft) tall,'],
-    ["The Moon's orbit around Earth has"],
-    ["The smooth Borealis basin in the Northern Hemisphere covers 40%"]
-]
 
-gr.Interface.load("huggingface/EleutherAI/gpt-j-6B", 
-                inputs=gr.inputs.Textbox(lines=4, label="Input Text"),
-                title=title,description=description,
-                examples=examples,
-                enable_queue=True).launch()
+title_ = "AIBG"
+description_ = " Enter 500 words blog "
+output_text = gr.outputs.Textbox()
+iface=gr.Interface(ai_text,"textbox", output_text, title=title_,description=description_)#.launch()
+iface.launch()