Spaces:
Sleeping
Sleeping
Enrico Shippole
commited on
Commit
·
d057fcb
1
Parent(s):
17d1f44
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(
|
|
|
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(
|