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