bluenevus commited on
Commit
1df2592
·
verified ·
1 Parent(s): 5ee7955

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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-3B-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
 
 
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