thong0710 commited on
Commit
e06abba
·
verified ·
1 Parent(s): ab41517
Files changed (1) hide show
  1. app/safety_check.py +1 -1
app/safety_check.py CHANGED
@@ -26,7 +26,7 @@ blip_processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning
26
  blip_model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
27
 
28
  # Load mô hình zero-shot nhẹ
29
- classifier = pipeline("zero-shot-classification", model="MoritzLaurer/distilbert-zero-shot-v1")
30
 
31
 
32
  # ========================== TEXT (Prompt) ==========================
 
26
  blip_model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")
27
 
28
  # Load mô hình zero-shot nhẹ
29
+ classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
30
 
31
 
32
  # ========================== TEXT (Prompt) ==========================