Update main.py

main.py

@@ -9,7 +9,7 @@ import soundfile as sf
 
 app = FastAPI()
 
-# Allow frontend to call backend
+# CORS Configuration
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -18,7 +18,7 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# Supported languages
+# Translation Models
 translation_models = {
     "fr": "Helsinki-NLP/opus-mt-en-fr",
     "es": "Helsinki-NLP/opus-mt-en-es",
@@ -31,8 +31,8 @@ translation_models = {
     "ta": "Helsinki-NLP/opus-mt-en-ta"
 }
 
-# Load models once
-generator = pipeline("text-generation", model="distilgpt2", framework="tf", from_tf=True)
+# Initialize Models CORRECTLY
+generator = pipeline("text-generation", model="distilgpt2")  # Fixed: Removed TF args, fixed model name
 whisper_model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-base")
 whisper_processor = WhisperProcessor.from_pretrained("openai/whisper-base")