Spaces:
Sleeping
Sleeping
fix: id
Browse files
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
|