Enrico Shippole commited on
Commit
d057fcb
1 Parent(s): 17d1f44

Add initial gradio setup

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, model):
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(