Spaces:
Sleeping
Sleeping
Fix again
Browse files- 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="
|
| 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) ==========================
|