Update app.py

app.py

@@ -27,7 +27,7 @@ Segment: Any = None
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 COMPUTE_TYPE = "float16" if device == "cuda" else "float32"
-BATCH_SIZE = 16
+BATCH_SIZE = 4
 token = os.environ.get("HF_TOKEN")
 global_diarizer = None
 def load_pyannote_pipeline():
@@ -46,7 +46,7 @@ def load_pyannote_pipeline():
     except Exception as e:
         print(f"Error loading pyannote pipeline dynamically: {type(e).__name__}: {e}. Diarization will be skipped.")
         return None
-model_name = "large-v1"
+model_name = "large-v2"
 ALIGN_MODEL_MAP = {
     "ur": "kingabzpro/wav2vec2-large-xls-r-300m-Urdu"}
 global_align_model_cache = {}