atwine
/

translation-en-lug-v8

Text2Text Generation

Inference Endpoints

Model card Files Files and versions Community

atwine commited on Oct 16, 2023

Commit

6f9e0c8

•

1 Parent(s): ed463ac

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -3,7 +3,7 @@ from transformers import AutoModelForSeq2SeqLM, AutoTokenizer, pipeline
 import torch
 class EndpointHandler:
-    def __init__(self, path="atwine/translation-en-lug-v6"):
         # Load model and processor from path
         self.model = AutoModelForSeq2SeqLM.from_pretrained(path)
         self.tokenizer = AutoTokenizer.from_pretrained(path)

 import torch
 class EndpointHandler:
+    def __init__(self, path=""):
         # Load model and processor from path
         self.model = AutoModelForSeq2SeqLM.from_pretrained(path)
         self.tokenizer = AutoTokenizer.from_pretrained(path)