nizar-sayad
/

opt-350m-HorrorCreepyPasta

Text Generation

Generated from Trainer

text-generation-inference

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

nizar-sayad commited on Apr 17, 2023

Commit

9b01ad2

·

1 Parent(s): 34bb7b3

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -4,7 +4,7 @@ import torch
 class EndpointHandler():
-    def __init__(self, path=""):
         # load model and processor from path
         self.model =  AutoModelForCausalLM.from_pretrained(path)
         self.tokenizer = AutoTokenizer.from_pretrained(path)

 class EndpointHandler():
+    def __init__(self, path="."):
         # load model and processor from path
         self.model =  AutoModelForCausalLM.from_pretrained(path)
         self.tokenizer = AutoTokenizer.from_pretrained(path)