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

Update src/app/model.py

Browse files
Files changed (1) hide show
  1. 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