Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 from faster_whisper import WhisperModel
 import logging
@@ -40,14 +41,15 @@ def text_to_srt(text):
     with open(temp_file_path, 'w', encoding='utf-8') as file:
         file.write(srt_content)
     return temp_file_path
-
+    
 # Fonction pour formater des secondes en timestamp
 def format_timestamp(seconds):
     hours = int(seconds // 3600)
     minutes = int((seconds % 3600) // 60)
     seconds_remainder = seconds % 60
     return f"{hours:02d}:{minutes:02d}:{seconds_remainder:06.3f}"
-
+    
+@spaces.GPU
 # Fonction de traduction de texte
 def translate_text(text, source_language_code, target_language_code):
     model_name = f"Helsinki-NLP/opus-mt-{source_language_code}-{target_language_code}"
@@ -61,7 +63,8 @@ def translate_text(text, source_language_code, target_language_code):
     translated = model.generate(**tokenizer(text, return_tensors="pt", padding=True, truncation=True, max_length=512))
     translated_text = tokenizer.decode(translated[0], skip_special_tokens=True)
     return translated_text
-
+    
+@spaces.GPU
 # Fonction pour traduire un fichier SRT
 def translate_srt(input_file_path, source_language_code, target_language_code, progress=None):
     subs = pysrt.open(input_file_path)
@@ -75,7 +78,8 @@ def translate_srt(input_file_path, source_language_code, target_language_code, p
     translated_srt_path = input_file_path.replace(".srt", f"_{target_language_code}.srt")
     pysrt.SubRipFile(translated_subs).save(translated_srt_path)
     return translated_srt_path
-
+    
+@spaces.GPU
 # Fonction pour transcrire l'audio d'une vidéo en texte
 def transcribe(audio_file_path, model_size="base"):
     device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -87,7 +91,8 @@ def transcribe(audio_file_path, model_size="base"):
         for segment in segments
     ]
     return "\n".join(transcription_with_timestamps)
-
+    
+@spaces.GPU
 # Fonction pour ajouter des sous-titres à une vidéo
 def add_subtitle_to_video(input_video, subtitle_file, subtitle_language, soft_subtitle=False):
     video_input_stream = ffmpeg.input(input_video)