Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 from transformers import pipeline
 
 # Load pipelines for Canary ASR, LLama3 QA, and VITS TTS
-asr_pipeline = pipeline("automatic-speech-recognition", model="canary/asr-small-librispeech", device=0)
+asr_pipeline = pipeline("automatic-speech-recognition", model="nvidia/canary-1b", device=0)
 qa_pipeline = pipeline("question-answering", model="LLAMA/llama3-base-qa", tokenizer="LLAMA/llama3-base-qa")
 tts_pipeline = pipeline("text-to-speech", model="patrickvonplaten/vits-large", device=0)
 
@@ -34,7 +34,7 @@ def ai_assistant(audio_input):
 if __name__ == "__main__":
     # Create a Gradio interface
     gr.Interface(ai_assistant,
-                 inputs=gr.inputs.Audio(capture= capture_audio, label="Speak Here"),
+                 inputs=gr.inputs.Audio(capture=capture_audio, label="Speak Here"),
                  outputs=gr.outputs.Audio(type="audio", label="Assistant's Response"),
                  title="AI Assistant",
-                 description="An AI Assistant that answers questions based on your speech input.").launch(inbrowser=True)
+                 description="An AI Assistant that answers questions based on your speech input.").launch()