sitammeur commited on
Commit
d118df2
·
verified ·
1 Parent(s): dd98ef8

Update src/app/model.py

Browse files
Files changed (1) hide show
  1. 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 tokenizer.
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.