Upload folder using huggingface_hub

hate_speech_demo.py

@@ -520,7 +520,10 @@ def get_contextual_rating(contextual_api, user_input):
     elif "caution" in response_text.lower() or "warning" in response_text.lower():
         safety_level = "warning"
     
-    return response_text, retrieval_text, safety_level
+    # Format the response text to separate rating, category, and explanation
+    formatted_response = format_contextual_rating(response_text)
+    
+    return formatted_response, retrieval_text, safety_level
 
 # LlamaGuard rating
 LLAMA_HARM_CATEGORIES = {