carson-together commited on
Commit
146454c
1 Parent(s): dbb601f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -45,8 +45,8 @@ def infer(
45
  if prompt=="":
46
  prompt = " "
47
  my_post_dict = {
48
- #"model": "togethercomputer/GPT-JT-6B-v1",
49
- "model":"EleutherAI/gpt-j-6b",
50
  "prompt": prompt,
51
  "top_p": top_p,
52
  "top_k": top_k,
 
45
  if prompt=="":
46
  prompt = " "
47
  my_post_dict = {
48
+ "model": "togethercomputer/GPT-JT-6B-v1",
49
+ #"model":"EleutherAI/gpt-j-6b",
50
  "prompt": prompt,
51
  "top_p": top_p,
52
  "top_k": top_k,