shaoyent commited on
Commit
71a4aa9
1 Parent(s): 46d3b2c
Files changed (1) hide show
  1. app.py +5 -15
app.py CHANGED
@@ -1,27 +1,18 @@
1
- # In[]:
2
- import sys
3
  import os
4
  import cv2
5
  import gradio as gr
6
- from PIL import Image
7
  import numpy as np
 
 
8
 
 
9
  from torch.nn.utils.rnn import pad_sequence
10
- from transformers import BridgeTowerForImageAndTextRetrieval, BridgeTowerProcessor
11
 
12
  from bridgetower_custom import BridgeTowerTextFeatureExtractor, BridgeTowerForITC
13
 
14
- import pickle
15
- from tqdm import tqdm
16
- from PIL import Image
17
-
18
- import torch
19
- import re
20
- import urllib.parse
21
  import faiss
22
-
23
  import webvtt
24
- import json
25
 
26
  from pytube import YouTube
27
  from youtube_transcript_api import YouTubeTranscriptApi
@@ -262,8 +253,7 @@ def get_video_id_from_url(video_url):
262
  return url.path.split('/')[2]
263
  if url.path[:3] == '/v/':
264
  return url.path.split('/')[2]
265
-
266
-
267
  return None
268
 
269
 
 
 
 
1
  import os
2
  import cv2
3
  import gradio as gr
 
4
  import numpy as np
5
+ import json
6
+ import pickle
7
 
8
+ import torch
9
  from torch.nn.utils.rnn import pad_sequence
10
+ from transformers import BridgeTowerProcessor
11
 
12
  from bridgetower_custom import BridgeTowerTextFeatureExtractor, BridgeTowerForITC
13
 
 
 
 
 
 
 
 
14
  import faiss
 
15
  import webvtt
 
16
 
17
  from pytube import YouTube
18
  from youtube_transcript_api import YouTubeTranscriptApi
 
253
  return url.path.split('/')[2]
254
  if url.path[:3] == '/v/':
255
  return url.path.split('/')[2]
256
+
 
257
  return None
258
 
259