Update app.py

app.py

@@ -10,13 +10,13 @@ from transformers import (
     AutoModelForSpeechSeq2Seq,
     TrainingArguments,
     Trainer,
-    DataCollatorSpeechSeq2Seq,  # ✅ Fix: Use correct data collator
+    DataCollatorForSeq2Seq,  # ✅ Fix: Use correct data collator
 )
 
 # ================================
 # 1️⃣ Authenticate with Hugging Face Hub (Securely)
 # ================================
-HF_TOKEN = os.getenv("hf_token")  # Ensure it's set in Hugging Face Spaces Secrets
+HF_TOKEN = os.getenv("HF_TOKEN")  # Ensure it's set in Hugging Face Spaces Secrets
 
 if HF_TOKEN is None:
     raise ValueError("❌ Hugging Face API token not found. Please set it in Secrets.")
@@ -117,7 +117,7 @@ training_args = TrainingArguments(
 )
 
 # ✅ FIX: Use correct Data Collator
-data_collator = DataCollatorSpeechSeq2Seq(processor=processor, model=model, return_tensors="pt")
+data_collator = DataCollatorForSeq2Seq(tokenizer=processor.tokenizer, model=model, return_tensors="pt")
 
 # Define Trainer
 trainer = Trainer(