bigPear commited on
Commit
0076ec7
·
1 Parent(s): 81d6669

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ sys.path.append("src")
5
  from src import load_pretrained, ModelArguments
6
 
7
  # if __name__ == "__main__":
8
- model_args = ModelArguments(checkpoint_dir="temp")
9
  model, tokenizer = load_pretrained(model_args)
10
  model = model.half().cuda()
11
  model = model.eval()
 
5
  from src import load_pretrained, ModelArguments
6
 
7
  # if __name__ == "__main__":
8
+ model_args = ModelArguments(checkpoint_dir="combined")
9
  model, tokenizer = load_pretrained(model_args)
10
  model = model.half().cuda()
11
  model = model.eval()