Update app.py
Browse files
app.py
CHANGED
|
@@ -48,8 +48,8 @@ def load_models():
|
|
| 48 |
classification_model.eval()
|
| 49 |
|
| 50 |
# 2. Load text generation model
|
| 51 |
-
|
| 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,
|