Spaces:
Runtime error
Runtime error
Update src/app/model.py
Browse files- src/app/model.py +1 -1
src/app/model.py
CHANGED
@@ -4,7 +4,7 @@ import sys
|
|
4 |
from dotenv import load_dotenv
|
5 |
from typing import Any
|
6 |
import torch
|
7 |
-
from transformers import AutoModel, AutoTokenizer
|
8 |
|
9 |
# Local imports
|
10 |
from src.logger import logging
|
|
|
4 |
from dotenv import load_dotenv
|
5 |
from typing import Any
|
6 |
import torch
|
7 |
+
from transformers import AutoModel, AutoTokenizer, AutoProcessor
|
8 |
|
9 |
# Local imports
|
10 |
from src.logger import logging
|