upload

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 
+from utils import get_youtube
 from model import summarize_video
 
 root_dir = '/home/user/app/video'
@@ -20,12 +21,7 @@ with demo:
                     <h1 style='text-align: center'>Video Summarization</h1>
                 </div>
             ''')
-    with gr.Column():
-        gr.Markdown('''
-            <div class="center">
-                <img src="https://user-images.githubusercontent.com/33136532/229133078-22cb84d6-b120-4a72-b1cf-b4b3ea47ed7d.png" width="500" height="300">
-            </div>
-        ''')
+    # with gr.Column():z
     with gr.Row():
             gr.Markdown('''
             ### Summarize video

model.py

@@ -13,7 +13,7 @@ from transformers import pipeline, BertTokenizer, BertForNextSentencePrediction
 import torch 
 import whisper
 
-from utils import convert_sample_rate, two_chnnel_to_one_channel, convert_sample_rate
+from utils import two_chnnel_to_one_channel, convert_sample_rate
 
 subprocess.run(['apt-get', '-y', 'install', 'imagemagick'])