wakeupmh commited on
Commit
7addc00
·
1 Parent(s): beba082
Files changed (1) hide show
  1. services/model_handler.py +1 -1
services/model_handler.py CHANGED
@@ -12,7 +12,7 @@ MODEL_PATH = "google/flan-t5-small"
12
  # Personnalized class for local models
13
  class LocalHuggingFaceModel(Model):
14
  def __init__(self, model, tokenizer, max_length=512):
15
- super().__init__()
16
  self.model = model
17
  self.tokenizer = tokenizer
18
  self.max_length = max_length
 
12
  # Personnalized class for local models
13
  class LocalHuggingFaceModel(Model):
14
  def __init__(self, model, tokenizer, max_length=512):
15
+ super().__init__(id="local-huggingface")
16
  self.model = model
17
  self.tokenizer = tokenizer
18
  self.max_length = max_length