Update src/app/model.py
Browse files- src/app/model.py +1 -1
src/app/model.py
CHANGED
@@ -3,8 +3,8 @@ import sys
|
|
3 |
import subprocess
|
4 |
from typing import Optional
|
5 |
from PIL import Image
|
6 |
-
from transformers import AutoProcessor, AutoModelForCausalLM
|
7 |
import spaces
|
|
|
8 |
|
9 |
# Local imports
|
10 |
from src.logger import logging
|
|
|
3 |
import subprocess
|
4 |
from typing import Optional
|
5 |
from PIL import Image
|
|
|
6 |
import spaces
|
7 |
+
from transformers import AutoProcessor, AutoModelForCausalLM
|
8 |
|
9 |
# Local imports
|
10 |
from src.logger import logging
|