Spaces:
Runtime error
Runtime error
Update kaggle_florence_gpu_2.py
Browse files- kaggle_florence_gpu_2.py +1 -1
kaggle_florence_gpu_2.py
CHANGED
@@ -15,7 +15,7 @@ import pickle
|
|
15 |
os.chdir("/kaggle/florence-sam-kaggle")
|
16 |
sys.path.append("/kaggle/florence-sam-kaggle")
|
17 |
from utils.video import generate_unique_name, create_directory, delete_directory
|
18 |
-
from utils.
|
19 |
FLORENCE_DETAILED_CAPTION_TASK, \
|
20 |
FLORENCE_CAPTION_TO_PHRASE_GROUNDING_TASK, FLORENCE_OPEN_VOCABULARY_DETECTION_TASK
|
21 |
from utils.modes import IMAGE_INFERENCE_MODES, IMAGE_OPEN_VOCABULARY_DETECTION_MODE, \
|
|
|
15 |
os.chdir("/kaggle/florence-sam-kaggle")
|
16 |
sys.path.append("/kaggle/florence-sam-kaggle")
|
17 |
from utils.video import generate_unique_name, create_directory, delete_directory
|
18 |
+
from utils.florencegpu2 import load_florence_model, run_florence_inference, \
|
19 |
FLORENCE_DETAILED_CAPTION_TASK, \
|
20 |
FLORENCE_CAPTION_TO_PHRASE_GROUNDING_TASK, FLORENCE_OPEN_VOCABULARY_DETECTION_TASK
|
21 |
from utils.modes import IMAGE_INFERENCE_MODES, IMAGE_OPEN_VOCABULARY_DETECTION_MODE, \
|