Guinnessgshep commited on
Commit
5bba6f5
1 Parent(s): 424e8fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
- title = "tory Generator"
4
 
5
  # gpt-neo-2.7B gpt-j-6B
6
 
@@ -28,9 +28,9 @@ def send(text1,context,text2):
28
  with gr.Blocks() as demo:
29
  textbox = gr.Textbox(placeholder="Type here and press enter...", lines=4)
30
  btn = gr.Button("Generate")
31
- context = gr.Slider(value=200,label="Truncate input text length (AI's memory)",minimum=1,maximum=500)
32
  the_model = gr.Dropdown(choices=['gpt2','gpt2-medium','gpt2-large','gpt2-xl','EleutherAI/gpt-neo-2.7B','EleutherAI/gpt-j-6B'],value = 'gpt2',label="Choose model")
33
- max_length = gr.Slider(value=20,label="Max Generate Length",minimum=1,maximum=50)
34
  temperature = gr.Slider(value=0.9,label="Temperature",minimum=0.0,maximum=1.0,step=0.05)
35
  repetition_penalty = gr.Slider(value=1.5,label="Repetition penalty",minimum=0.2,maximum=2,step=0.1)
36
  output1 = gr.Textbox(lines=4,label='1')
 
1
  import gradio as gr
2
  from transformers import pipeline
3
+ title = "story Generator"
4
 
5
  # gpt-neo-2.7B gpt-j-6B
6
 
 
28
  with gr.Blocks() as demo:
29
  textbox = gr.Textbox(placeholder="Type here and press enter...", lines=4)
30
  btn = gr.Button("Generate")
31
+ context = gr.Slider(value=2000,label="Truncate input text length (AI's memory)",minimum=1,maximum=50000)
32
  the_model = gr.Dropdown(choices=['gpt2','gpt2-medium','gpt2-large','gpt2-xl','EleutherAI/gpt-neo-2.7B','EleutherAI/gpt-j-6B'],value = 'gpt2',label="Choose model")
33
+ max_length = gr.Slider(value=20,label="Max Generate Length",minimum=1,maximum=50000)
34
  temperature = gr.Slider(value=0.9,label="Temperature",minimum=0.0,maximum=1.0,step=0.05)
35
  repetition_penalty = gr.Slider(value=1.5,label="Repetition penalty",minimum=0.2,maximum=2,step=0.1)
36
  output1 = gr.Textbox(lines=4,label='1')