Enrico Shippole commited on
Commit
82872e3
1 Parent(s): 2988fa3

Add initial gradio setup

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -3,7 +3,7 @@ from transformers import AutoTokenizer
3
  from palm_rlhf_pytorch import PaLM
4
  import gradio as gr
5
 
6
- def generate(prompt, seq_len, temperature, filter_thres):
7
  device = torch.device("cpu")
8
 
9
  model = PaLM(
@@ -38,9 +38,9 @@ iface = gr.Interface(
38
  description="Open-source PaLM demo.",
39
  inputs="text",
40
  outputs="text",
41
- seq_len=gr.Slider(minimum=1, maximum=512, step=1, default=128, label="Sequence Length"),
42
- temperature=gr.Slider(minimum=0.0, maximum=1.0, step=0.01, default=0.9, label="Temperature"),
43
- filter_thres=gr.Slider(minimum=0.0, maximum=1.0, step=0.01, default=0.9, label="Filter Threshold"),
44
  )
45
 
46
  iface.launch()
 
3
  from palm_rlhf_pytorch import PaLM
4
  import gradio as gr
5
 
6
+ def generate(prompt, seq_len=128, temperature=0.8, filter_thres=0.9):
7
  device = torch.device("cpu")
8
 
9
  model = PaLM(
 
38
  description="Open-source PaLM demo.",
39
  inputs="text",
40
  outputs="text",
41
+ # seq_len=gr.Slider(minimum=1, maximum=8192, step=1, default=32, label="Sequence Length"),
42
+ # temperature=gr.Slider(minimum=0.0, maximum=1.0, step=0.01, default=0.8, label="Temperature"),
43
+ # filter_thres=gr.Slider(minimum=0.0, maximum=1.0, step=0.01, default=0.9, label="Filter Threshold"),
44
  )
45
 
46
  iface.launch()