supersolar commited on
Commit
87ce726
1 Parent(s): 9610729

Update kaggle_gpu_2.py

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