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 |
|