supersolar commited on
Commit
3a235a7
·
verified ·
1 Parent(s): 0eaeee2

Update kaggle_florence_gpu_2.py

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