Update main.py

main.py

@@ -32,7 +32,7 @@ translation_models = {
 }
 
 # Load models once
-generator = pipeline("text-generation", model="distilgpt2")  # Fixed line
+generator = pipeline("text-generation", model="distilgpt2", framework="tf", from_tf=True)
 whisper_model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-base")
 whisper_processor = WhisperProcessor.from_pretrained("openai/whisper-base")