Locutusque commited on
Commit
c3ce568
1 Parent(s): dec3cbd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -29,14 +29,14 @@ def generate_text(prompt, temperature, top_p, top_k, repetition_penalty, max_len
29
  iface = gr.Interface(
30
  fn=generate_text,
31
  inputs=[
32
- gr.inputs.Textbox(label="Prompt", lines=2, placeholder="Type a prompt..."),
33
- gr.inputs.Slider(minimum=0.1, maximum=2.0, step=0.01, default=0.8, label="Temperature"),
34
- gr.inputs.Slider(minimum=0.0, maximum=1.0, step=0.01, default=0.95, label="Top p"),
35
- gr.inputs.Slider(minimum=0, maximum=100, step=1, default=40, label="Top k"),
36
- gr.inputs.Slider(minimum=1.0, maximum=2.0, step=0.01, default=1.10, label="Repetition Penalty"),
37
- gr.inputs.Slider(minimum=5, maximum=4096, step=5, default=1024, label="Max Length")
38
  ],
39
- outputs=gr.outputs.Textbox(label="Generated Text"),
40
  title="Text Completion Model",
41
  description="Try out the Mistral-7B model for free! Note this is the pretrained model and is not fine-tuned for instruction."
42
  )
 
29
  iface = gr.Interface(
30
  fn=generate_text,
31
  inputs=[
32
+ gr.Textbox(label="Prompt", lines=2, placeholder="Type a prompt..."),
33
+ gr.Slider(minimum=0.1, maximum=2.0, step=0.01, default=0.8, label="Temperature"),
34
+ gr.Slider(minimum=0.0, maximum=1.0, step=0.01, default=0.95, label="Top p"),
35
+ gr.Slider(minimum=0, maximum=100, step=1, default=40, label="Top k"),
36
+ gr.Slider(minimum=1.0, maximum=2.0, step=0.01, default=1.10, label="Repetition Penalty"),
37
+ gr.Slider(minimum=5, maximum=4096, step=5, default=1024, label="Max Length")
38
  ],
39
+ outputs=gr.Textbox(label="Generated Text"),
40
  title="Text Completion Model",
41
  description="Try out the Mistral-7B model for free! Note this is the pretrained model and is not fine-tuned for instruction."
42
  )