Spaces:
Running
Running
Update kaggle_florence_gpu_1.py
Browse files- 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)
|