krplatz
/

gemma-med3-9b

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

krplatz commited on Sep 3

Commit

948e8bf

•

1 Parent(s): 7d810d6

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -1,7 +1,7 @@
 from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
 import torch
-class Gemma2Handler:
     def __init__(self, model_dir):
         # Load the tokenizer
         self.tokenizer = AutoTokenizer.from_pretrained(model_dir)

 from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
 import torch
+class EndpointHandler:
     def __init__(self, model_dir):
         # Load the tokenizer
         self.tokenizer = AutoTokenizer.from_pretrained(model_dir)