Abijith commited on
Commit
936cd6e
·
verified ·
1 Parent(s): 5bec6c1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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 = "meta-llama/Llama-3.2-1B-Instruct"
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",