commit

app.py

@@ -9,6 +9,8 @@ from whisper.utils import write_vtt
 
 model = whisper.load_model("medium")
 
+title = "Add Text/Caption to your YouTube Shorts - MultiLingual"
+
 def video2mp3(video_file, output_ext="mp3"):
     filename, ext = os.path.splitext(video_file)
     subprocess.call(["ffmpeg", "-y", "-i", video_file, f"{filename}.{output_ext}"], 
@@ -25,7 +27,7 @@ def translate(input_video):
     translate_options = dict(task="translate", **options)
     result = model.transcribe(audio_file,**translate_options)
 
-    output_dir = '/content/'
+    output_dir = ''
     audio_path = audio_file.split(".")[0]
 
     with open(os.path.join(output_dir, audio_path + ".vtt"), "w") as vtt:
@@ -38,21 +40,12 @@ def translate(input_video):
 
     return output_video
 
-
-
-title = "Add Caption to your Videos - MultiLingual"
-
 block = gr.Blocks()
-
 with block:
 
     with gr.Group():
-        with gr.Box(): 
-
-           
-
+        with gr.Box():
             with gr.Row().style():
-               
                 inp_video = gr.Video(
                     label="Input Video",
                     type="filepath",
@@ -67,7 +60,6 @@ with block:
 
         
         btn.click(translate, inputs=[inp_video], outputs=[op_video])
- 
         gr.HTML('''
         <div class="footer">
                     <p>Powered by <a href="https://openai.com/" style="text-decoration: underline;" target="_blank">OpenAI</a> - Developed by <a href="https://github.com/Isaakkamau" style="text-decoration: underline;" target="_blank">+254704205553</a>
@@ -75,4 +67,4 @@ with block:
         </div>
         ''')
 
-block.launch(debug = True )
+block.launch(enable_queue = True)

requirements.txt

@@ -1,5 +1,5 @@
-gradio
-streamlit==1.22.0
+tensorflow
+openai-whisper==20230117
 gTTS==2.2.2
 googletrans==3.1.0a0
 openai-whisper==20230117