Commit
•
efc6f35
1
Parent(s):
13fdbce
Update handler.py
Browse files- 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 |
|