Spaces:
Runtime error
Runtime error
supersolar
commited on
Commit
•
87ce726
1
Parent(s):
9610729
Update kaggle_gpu_2.py
Browse files- kaggle_gpu_2.py +1 -1
kaggle_gpu_2.py
CHANGED
@@ -10,7 +10,7 @@ import torch
|
|
10 |
from PIL import Image
|
11 |
from tqdm import tqdm
|
12 |
from utils.video import generate_unique_name, create_directory, delete_directory
|
13 |
-
from utils.
|
14 |
FLORENCE_DETAILED_CAPTION_TASK, \
|
15 |
FLORENCE_CAPTION_TO_PHRASE_GROUNDING_TASK, FLORENCE_OPEN_VOCABULARY_DETECTION_TASK
|
16 |
from utils.modes import IMAGE_INFERENCE_MODES, IMAGE_OPEN_VOCABULARY_DETECTION_MODE, \
|
|
|
10 |
from PIL import Image
|
11 |
from tqdm import tqdm
|
12 |
from utils.video import generate_unique_name, create_directory, delete_directory
|
13 |
+
from utils.florencegpu2 import load_florence_model, run_florence_inference, \
|
14 |
FLORENCE_DETAILED_CAPTION_TASK, \
|
15 |
FLORENCE_CAPTION_TO_PHRASE_GROUNDING_TASK, FLORENCE_OPEN_VOCABULARY_DETECTION_TASK
|
16 |
from utils.modes import IMAGE_INFERENCE_MODES, IMAGE_OPEN_VOCABULARY_DETECTION_MODE, \
|