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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -45,7 +45,8 @@ def infer(
45
  if prompt=="":
46
  prompt = " "
47
  my_post_dict = {
48
- "model": "togethercomputer/GPT-JT-6B-v1",
 
49
  "prompt": prompt,
50
  "top_p": top_p,
51
  "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,