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

Update src/app/model.py

Browse files
Files changed (1) hide show
  1. src/app/model.py +2 -1
src/app/model.py CHANGED
@@ -29,9 +29,10 @@ def load_model_and_tokenizer(model_name: str, device: str) -> Any:
29
  Returns:
30
  - model: The loaded model.
31
  - tokenizer: The loaded tokenizer.
 
32
  """
33
  try:
34
- # Load the model and tokenizer
35
  model = AutoModel.from_pretrained(
36
  model_name,
37
  trust_remote_code=True,
 
29
  Returns:
30
  - model: The loaded model.
31
  - tokenizer: The loaded tokenizer.
32
+ - processor: The loaded processor.
33
  """
34
  try:
35
+ # Load the model, tokenizer and processor
36
  model = AutoModel.from_pretrained(
37
  model_name,
38
  trust_remote_code=True,