Update app.py
Browse files
app.py
CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
|
|
2 |
from transformers import pipeline
|
3 |
|
4 |
# Create pipelines for ASR, QA, and TTS
|
5 |
-
asr_pipeline = pipeline("automatic-speech-recognition", model="canary
|
6 |
qa_pipeline = pipeline("question-answering", model="LLAMA/llama3-base-qa", tokenizer="LLAMA/llama3-base-qa")
|
7 |
tts_pipeline = pipeline("text-to-speech", model="patrickvonplaten/vits-large", device=0) # Adjust device based on your hardware
|
8 |
|
|
|
2 |
from transformers import pipeline
|
3 |
|
4 |
# Create pipelines for ASR, QA, and TTS
|
5 |
+
asr_pipeline = pipeline("automatic-speech-recognition", model="nvidia/canary-1b", device=0) # Adjust device based on your hardware
|
6 |
qa_pipeline = pipeline("question-answering", model="LLAMA/llama3-base-qa", tokenizer="LLAMA/llama3-base-qa")
|
7 |
tts_pipeline = pipeline("text-to-speech", model="patrickvonplaten/vits-large", device=0) # Adjust device based on your hardware
|
8 |
|