Update app.py

app.py

@@ -4,4 +4,25 @@ def greet(name):
     return "results " + name + "!!"
 
 iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+iface.launch()
+
+#When running locally, you won`t have access to this, so you can remove this part
+word_list_dataset = load_dataset("stabilityai/word-list", data_files="list.txt", use_auth_token=True)
+word_list = word_list_dataset["train"]['text']
+
+is_gpu_busy = False
+def infer(prompt):
+    global is_gpu_busy
+    samples = 4
+    steps = 50
+    scale = 7.5
+    #When running locally you can also remove this filter
+    for filter in word_list:
+        if re.search(rf"\b{filter}\b", prompt):
+            raise gr.Error("Unsafe content found. Please try again with different prompts.")
+        
+    #generator = torch.Generator(device=device).manual_seed(seed)
+    #print("Is GPU busy? ", is_gpu_busy)
+    images = []
+    #if(not is_gpu_busy):
+    #    is_gpu_busy = True