updated requirements, fixed decroators

README.md

@@ -4,7 +4,7 @@ emoji: 👀
 colorFrom: pink
 colorTo: green
 sdk: gradio
-sdk_version: 4.42.0
+sdk_version: 4.43.0
 app_file: app.py
 pinned: false
 ---

app.py

@@ -10,7 +10,7 @@ if torch.cuda.is_available():
 else:
     print("No CUDA GPUs available. Running on CPU.")
 
-@spaces.GPU(duration=180)
+# @spaces.GPU(duration=180)
 def transcribe_audio(audio_file, translate, model_size):
     language_segments, final_segments = process_audio(audio_file, translate=translate, model_size=model_size)
     

audio_processing.py

@@ -11,7 +11,7 @@ import time
 from difflib import SequenceMatcher
 hf_token = os.getenv("HF_TOKEN")
 
-CHUNK_LENGTH=5
+CHUNK_LENGTH=10
 OVERLAP=2
 import whisperx
 import torch
@@ -37,7 +37,7 @@ def process_audio(audio_file, translate=False, model_size="small"):
     try:
         device = "cuda" if torch.cuda.is_available() else "cpu"
         print(f"Using device: {device}")
-        compute_type = "float16"
+        compute_type = "float32"
         audio = whisperx.load_audio(audio_file)
         model = whisperx.load_model(model_size, device, compute_type=compute_type)
 

requirements.txt

@@ -5,7 +5,7 @@ pandas
 pyannote.audio
 pyperclip
 sentencepiece
-gradio
+gradio==4.43.0
 speechbrain==0.5.16
 --extra-index-url https://download.pytorch.org/whl/cu113
 torch