Update app.py

app.py

@@ -5,6 +5,7 @@ import gradio as gr
 
 MODEL_NAME = "Sagicc/whisper-large-v3-sr-combined"
 BATCH_SIZE = 8
+lang = "sr"
 
 device = 0 if torch.cuda.is_available() else "cpu"
 
@@ -38,7 +39,7 @@ def format_timestamp(seconds: float, always_include_hours: bool = False, decimal
 
 
 def transcribe(file, task, return_timestamps):
-    outputs = pipe(file, batch_size=BATCH_SIZE, language="sr", generate_kwargs={"task": task}, return_timestamps=return_timestamps)
+    outputs = pipe(file, batch_size=BATCH_SIZE, language=lang, generate_kwargs={"task": task}, return_timestamps=return_timestamps)
     text = outputs["text"]
     if return_timestamps:
         timestamps = outputs["chunks"]