Update app.py

app.py

@@ -3,9 +3,8 @@ import io
 import torch
 import uvicorn
 import spacy
-import subprocess  # Added for running ffmpeg commands
+import subprocess  # For running ffmpeg commands
 import pdfplumber
-# Removed: import moviepy.editor as mp
 import librosa
 import soundfile as sf
 import matplotlib.pyplot as plt
@@ -165,6 +164,7 @@ def fine_tune_cuad_model():
     )
 
     print("✅ Starting fine tuning on CUAD QA dataset...")
+    from transformers import Trainer  # Ensure Trainer is imported here
     trainer = Trainer(
         model=model,
         args=training_args,
@@ -185,9 +185,6 @@ def fine_tune_cuad_model():
 #    Load NLP Models       #
 #############################
 
-# ... (other imports and code remain unchanged)
-
-# Load NLP models
 try:
     try:
         nlp = spacy.load("en_core_web_sm")
@@ -211,9 +208,6 @@ try:
                               chunk_length_s=30,
                               device_map="auto" if torch.cuda.is_available() else "cpu")
 
-    # ... (rest of your model loading code remains unchanged)
-
-
     # Load or Fine Tune CUAD QA Model
     if os.path.exists("fine_tuned_legal_qa"):
         print("✅ Loading fine-tuned CUAD QA model from fine_tuned_legal_qa...")
@@ -747,3 +741,4 @@ if __name__ == "__main__":
     else:
         print("\n⚠️ Ngrok setup failed. API will only be available locally.\n")
     run()
+