Spaces:
Runtime error
Runtime error
Update src/app/model.py
Browse files- src/app/model.py +1 -1
src/app/model.py
CHANGED
@@ -20,7 +20,7 @@ access_token = os.environ.get("ACCESS_TOKEN")
|
|
20 |
|
21 |
def load_model_and_tokenizer(model_name: str, device: str) -> Any:
|
22 |
"""
|
23 |
-
Load the model and
|
24 |
|
25 |
Args:
|
26 |
- model_name (str): The name of the model to load.
|
|
|
20 |
|
21 |
def load_model_and_tokenizer(model_name: str, device: str) -> Any:
|
22 |
"""
|
23 |
+
Load the model, tokenizer and processor.
|
24 |
|
25 |
Args:
|
26 |
- model_name (str): The name of the model to load.
|