jamesesqueleto commited on
Commit
61c192c
·
1 Parent(s): 2469fe6

added spaces

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. requirements.txt +1 -0
app.py CHANGED
@@ -5,7 +5,7 @@ import pandas as pd
5
  from moviepy import VideoFileClip
6
  from moviepy.video.tools.subtitles import SubtitlesClip
7
  import os
8
-
9
  from moviepy.video.tools.subtitles import SubtitlesClip
10
  from moviepy.video.io.VideoFileClip import VideoFileClip
11
  from moviepy import CompositeVideoClip
@@ -111,7 +111,7 @@ def convert_audio_to_mono_16khz(input_path, output_path):
111
  print(f"Audio converted successfully to mono, 16kHz at: {output_path}")
112
  except Exception as e:
113
  print(f"Error converting audio: {e}")
114
-
115
  def subtitle_video(input_file):
116
  # ------------------------------------------------------------------------------------------------------------------------------
117
  # Params:
 
5
  from moviepy import VideoFileClip
6
  from moviepy.video.tools.subtitles import SubtitlesClip
7
  import os
8
+ import spaces
9
  from moviepy.video.tools.subtitles import SubtitlesClip
10
  from moviepy.video.io.VideoFileClip import VideoFileClip
11
  from moviepy import CompositeVideoClip
 
111
  print(f"Audio converted successfully to mono, 16kHz at: {output_path}")
112
  except Exception as e:
113
  print(f"Error converting audio: {e}")
114
+ @spaces.GPU
115
  def subtitle_video(input_file):
116
  # ------------------------------------------------------------------------------------------------------------------------------
117
  # Params:
requirements.txt CHANGED
@@ -29,4 +29,5 @@ hydra-core
29
  pytorch-lightning
30
  flash-attn
31
  cuda-python
 
32
  git+https://github.com/NVIDIA/NeMo.git@main#egg=nemo_toolkit[asr]
 
29
  pytorch-lightning
30
  flash-attn
31
  cuda-python
32
+ spaces
33
  git+https://github.com/NVIDIA/NeMo.git@main#egg=nemo_toolkit[asr]