Spaces:
Runtime error
Runtime error
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()
|