Spaces:
Runtime error
Runtime error
Update src/app/model.py
Browse files- src/app/model.py +1 -1
src/app/model.py
CHANGED
|
@@ -15,7 +15,7 @@ from src.exception import CustomExceptionHandling
|
|
| 15 |
load_dotenv()
|
| 16 |
|
| 17 |
# Access token for using the model
|
| 18 |
-
access_token = "
|
| 19 |
|
| 20 |
|
| 21 |
def load_model_and_tokenizer(model_name: str, device: str) -> Any:
|
|
|
|
| 15 |
load_dotenv()
|
| 16 |
|
| 17 |
# Access token for using the model
|
| 18 |
+
access_token = os.environ.get("ACCESS_TOKEN")
|
| 19 |
|
| 20 |
|
| 21 |
def load_model_and_tokenizer(model_name: str, device: str) -> Any:
|