Update app.py

app.py

@@ -2,10 +2,11 @@ from transformers import pipeline
 import gradio as gr
 
 # Load pre-trained models
-stt_model = pipeline("automatic-speech-recognition", model="openai/whisper-base")
-nlp_model = pipeline("text-generation", model="gpt2")
+stt_model = pipeline("automatic-speech-recognition", model="openai/whisper-tiny")
+nlp_model = pipeline("text-generation", model="sshleifer/tiny-gpt2")
 tts_model = pipeline("text-to-speech", model="tts-coqui/coqui-tts-en")
 
+
 # Define a function to handle the workflow
 def conversation(audio):
     # Step 1: Convert speech to text