sitammeur commited on
Commit
45da847
·
verified ·
1 Parent(s): 284c0cd

Update src/app/model.py

Browse files
Files changed (1) hide show
  1. src/app/model.py +1 -1
src/app/model.py CHANGED
@@ -52,7 +52,7 @@ def load_model_and_tokenizer(model_name: str, device: str) -> Any:
52
  # Log the successful loading of the model and tokenizer
53
  logging.info("Model and tokenizer loaded successfully.")
54
 
55
- # Return the model and tokenizer
56
  return model, tokenizer, processor
57
 
58
  # Handle exceptions that may occur during model and tokenizer loading
 
52
  # Log the successful loading of the model and tokenizer
53
  logging.info("Model and tokenizer loaded successfully.")
54
 
55
+ # Return the model, tokenizer and processor
56
  return model, tokenizer, processor
57
 
58
  # Handle exceptions that may occur during model and tokenizer loading