saadkiet commited on
Commit
d0db16c
1 Parent(s): 88492ab

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -0
app.py ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from aitextgen import aitextgen
2
+ import gradio as gr
3
+
4
+
5
+ ai=aitextgen(model='EleutherAI/gpt-neo-1.3B',to_gpu=False)
6
+
7
+ def ai_text(Input):
8
+ generated_text = ai.generate_one(max_length = 500, prompt = Input, no_repeat_ngram_size = 3) #repetition_penalty = 1.9)
9
+ #print(type(generated_text))
10
+ return generated_text
11
+
12
+
13
+ title_ = "AI Blog Generation 😎"
14
+ description_ = " Enter few sentence and it will generate a 500 words blog 🤓"
15
+ output_text = gr.outputs.Textbox()
16
+ iface=gr.Interface(ai_text,"textbox", output_text, title=title_,description=description_)#.launch()
17
+ iface.launch()