Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -24,8 +24,7 @@ from huggingface_hub import login
|
|
24 |
# login(token=hf_token)
|
25 |
|
26 |
# Load the model and tokenizer
|
27 |
-
model_name = "
|
28 |
-
# model_name = "HuggingFaceTB/SmolLM2-360M-Instruct"
|
29 |
|
30 |
# Initialize classifier once for input guardrail
|
31 |
classifier = pipeline("zero-shot-classification",
|
|
|
24 |
# login(token=hf_token)
|
25 |
|
26 |
# Load the model and tokenizer
|
27 |
+
model_name = "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
|
|
|
28 |
|
29 |
# Initialize classifier once for input guardrail
|
30 |
classifier = pipeline("zero-shot-classification",
|