Spaces:
Running
on
Zero
Running
on
Zero
Update src/app/model.py
Browse files- 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
|
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
|