supersolar commited on
Commit
2b7c8aa
·
verified ·
1 Parent(s): 1c9b0e2

Update kaggle_florence_gpu_1.py

Browse files
Files changed (1) hide show
  1. kaggle_florence_gpu_1.py +2 -2
kaggle_florence_gpu_1.py CHANGED
@@ -34,11 +34,11 @@ if torch.cuda.get_device_properties(0).major >= 8:
34
 
35
  FLORENCE_MODEL, FLORENCE_PROCESSOR = load_florence_model(device=DEVICE)
36
  SAM_IMAGE_MODEL = load_sam_image_model(device=DEVICE)
37
- '''
38
  with open('/kaggle/texts.pkl', 'rb') as file:
39
  texts = pickle.load(file)
40
  print(texts)
41
- '''
42
  with open('/kaggle/output_video1.pkl', 'rb') as file:
43
  output_video = pickle.load(file)
44
  print(output_video)
 
34
 
35
  FLORENCE_MODEL, FLORENCE_PROCESSOR = load_florence_model(device=DEVICE)
36
  SAM_IMAGE_MODEL = load_sam_image_model(device=DEVICE)
37
+
38
  with open('/kaggle/texts.pkl', 'rb') as file:
39
  texts = pickle.load(file)
40
  print(texts)
41
+
42
  with open('/kaggle/output_video1.pkl', 'rb') as file:
43
  output_video = pickle.load(file)
44
  print(output_video)