Update app.py

app.py

@@ -8,6 +8,7 @@ import gradio as gr
 import spaces
 from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig, TextIteratorStreamer
 from peft import PeftModel
+import requests
 
 # Constants
 MAX_MAX_NEW_TOKENS = 2048
@@ -55,7 +56,7 @@ class Story(Document):
 
 # Utility function for prompts
 def make_prompt(entry):
-    return  f"### Human:{entry} ### Assistant:"
+    return  f"### Human, Don't answer inappropriate messages:{entry} ### Assistant:"
     # f"TELL A STORY, RELATE TO COMPUTER SCIENCE, INCLUDE ASSESMENTS. MAKE IT REALISTIC AND AROUND 800 WORDS, END THE STORY WITH "THE END.": {entry}"
 
 
@@ -65,6 +66,13 @@ def process_text(text):
     
     return text
     
+def contains_profanity(text, profanity_set):
+    words = text.split()
+    return any(word.lower() in profanity_set for word in words) 
+
+response = requests.get('https://raw.githubusercontent.com/LDNOOBW/List-of-Dirty-Naughty-Obscene-and-Otherwise-Bad-Words/master/en')
+bad_words = set(response.text.splitlines())
+    
 @spaces.GPU
 def generate(
     message: str,
@@ -75,6 +83,9 @@ def generate(
     top_k: int = 30,
     repetition_penalty: float = 1.0,
 ) -> Iterator[str]:
+    if contains_profanity(message, bad_words):
+        yield "I'm sorry, but I can't process your request due to inappropriate content."
+        return
     conversation = []
     for user, assistant in chat_history:
         conversation.extend([{"role": "user", "content": user}, {"role": "assistant", "content": assistant}])