Spaces:
Sleeping
Sleeping
Enrico Shippole
commited on
Commit
·
82872e3
1
Parent(s):
2988fa3
Add initial gradio setup
Browse files
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=
|
42 |
-
temperature=gr.Slider(minimum=0.0, maximum=1.0, step=0.01, default=0.
|
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()
|