ElAnon commited on
Commit
2232bc2
1 Parent(s): 68fe1d4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,13 +5,13 @@ import gradio as gr
5
  ai=aitextgen(model='EleutherAI/gpt-neo-1.3B',to_gpu=False) # EleutherAI/gpt-neo-2.7B EleutherAI/gpt-neo-1.3B
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_ = "AIBG"
14
- description_ = " Enter 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()
 
5
  ai=aitextgen(model='EleutherAI/gpt-neo-1.3B',to_gpu=False) # EleutherAI/gpt-neo-2.7B EleutherAI/gpt-neo-1.3B
6
 
7
  def ai_text(Input):
8
+ generated_text = ai.generate_one(max_length = 450, prompt = Input, no_repeat_ngram_size = 3) #repetition_penalty = 1.9)
9
  #print(type(generated_text))
10
  return generated_text
11
 
12
 
13
  title_ = "AIBG"
14
+ description_ = " Enter 450 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()