Update app.py
Browse files
app.py
CHANGED
|
@@ -30,7 +30,7 @@ whisper_processor = WhisperProcessor.from_pretrained(whisper_model_name)
|
|
| 30 |
whisper_model = WhisperForConditionalGeneration.from_pretrained(whisper_model_name).to(device)
|
| 31 |
|
| 32 |
# Load the Qwen model and tokenizer
|
| 33 |
-
qwen_model_name = "Qwen/Qwen2.5-
|
| 34 |
qwen_tokenizer = AutoTokenizer.from_pretrained(qwen_model_name, trust_remote_code=True)
|
| 35 |
qwen_model = AutoModelForCausalLM.from_pretrained(qwen_model_name, trust_remote_code=True).to(device)
|
| 36 |
|
|
|
|
| 30 |
whisper_model = WhisperForConditionalGeneration.from_pretrained(whisper_model_name).to(device)
|
| 31 |
|
| 32 |
# Load the Qwen model and tokenizer
|
| 33 |
+
qwen_model_name = "Qwen/Qwen2.5-1.5B-Instruct"
|
| 34 |
qwen_tokenizer = AutoTokenizer.from_pretrained(qwen_model_name, trust_remote_code=True)
|
| 35 |
qwen_model = AutoModelForCausalLM.from_pretrained(qwen_model_name, trust_remote_code=True).to(device)
|
| 36 |
|