Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -377,7 +377,7 @@ def vqa_answer(image, question):
|
|
| 377 |
out = vqa_model.generate(**inputs, max_new_tokens=50)
|
| 378 |
answer = vqa_processor.decode(out[0], skip_special_tokens=True)
|
| 379 |
|
| 380 |
-
|
| 381 |
beep = make_beep_sound()
|
| 382 |
tts = gTTS("Warning! Unsafe or inappropriate content detected.", lang="en")
|
| 383 |
speech_tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".mp3")
|
|
|
|
| 377 |
out = vqa_model.generate(**inputs, max_new_tokens=50)
|
| 378 |
answer = vqa_processor.decode(out[0], skip_special_tokens=True)
|
| 379 |
|
| 380 |
+
if not is_caption_safe(answer):
|
| 381 |
beep = make_beep_sound()
|
| 382 |
tts = gTTS("Warning! Unsafe or inappropriate content detected.", lang="en")
|
| 383 |
speech_tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".mp3")
|