Ahsen Khaliq commited on
Commit
02e7400
1 Parent(s): 602c097

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -63,9 +63,9 @@ def inference(video, text):
63
  print(f"Features: {video_features.shape}")
64
 
65
 
66
- search_query=text
67
- display_heatmap=False
68
- display_results_count=1
69
  # Encode and normalize the search query using CLIP
70
  with torch.no_grad():
71
  text_features = model.encode_text(clip.tokenize(search_query).to(device))
 
63
  print(f"Features: {video_features.shape}")
64
 
65
 
66
+ search_query=text
67
+ display_heatmap=False
68
+ display_results_count=1
69
  # Encode and normalize the search query using CLIP
70
  with torch.no_grad():
71
  text_features = model.encode_text(clip.tokenize(search_query).to(device))