jschwab21 commited on
Commit
1876e9f
·
verified ·
1 Parent(s): 36c8dbf

Update video_processing.py

Browse files
Files changed (1) hide show
  1. video_processing.py +1 -1
video_processing.py CHANGED
@@ -112,7 +112,7 @@ def analyze_scenes(video_path, scenes, description, batch_size=4):
112
  text_inputs = processor(text=[description] + negative_descriptions, return_tensors="pt", padding=True).to(device)
113
  text_features = model.get_text_features(**text_inputs).detach()
114
  positive_feature, negative_features = text_features[0], text_features[1:]
115
- print("Negative features shape:", negative_features.shape())
116
  video = VideoFileClip(video_path)
117
 
118
  for scene_num, (start_time, end_time) in enumerate(scenes):
 
112
  text_inputs = processor(text=[description] + negative_descriptions, return_tensors="pt", padding=True).to(device)
113
  text_features = model.get_text_features(**text_inputs).detach()
114
  positive_feature, negative_features = text_features[0], text_features[1:]
115
+ print("Negative features shape:", negative_features)
116
  video = VideoFileClip(video_path)
117
 
118
  for scene_num, (start_time, end_time) in enumerate(scenes):