Deepakvictor commited on
Commit
efc6f35
1 Parent(s): 13fdbce

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -5,7 +5,7 @@ from handler import EndpointHandler
5
 
6
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
7
  class EndpointHandler:
8
- def __init__(self, path=""):
9
  self.tokenizer = AutoTokenizer.from_pretrained(path).to(device)
10
  self.model = AutoModelForSeq2SeqLM.from_pretrained(path).to(device))
11
 
 
5
 
6
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
7
  class EndpointHandler:
8
+ def __init__(self, path="Deepakvictor/tan-ta"):
9
  self.tokenizer = AutoTokenizer.from_pretrained(path).to(device)
10
  self.model = AutoModelForSeq2SeqLM.from_pretrained(path).to(device))
11