TETSU0701 commited on
Commit
4f52637
·
verified ·
1 Parent(s): 0fc23c1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -48,8 +48,8 @@ def load_models():
48
  classification_model.eval()
49
 
50
  # 2. Load text generation model
51
- # llm_model_name = "Qwen/Qwen3-0.6B"
52
- llm_model_name = "Qwen/QwQ-32B"
53
  tokenizer = AutoTokenizer.from_pretrained(llm_model_name)
54
  llm_model = AutoModelForCausalLM.from_pretrained(
55
  llm_model_name,
 
48
  classification_model.eval()
49
 
50
  # 2. Load text generation model
51
+ llm_model_name = "Qwen/Qwen3-0.6B"
52
+ # llm_model_name = "Qwen/QwQ-32B"
53
  tokenizer = AutoTokenizer.from_pretrained(llm_model_name)
54
  llm_model = AutoModelForCausalLM.from_pretrained(
55
  llm_model_name,