Update app.py

app.py

@@ -9,7 +9,7 @@ generator = pipeline('text-generation', model='PygmalionAI/pygmalion-1.3b', retu
 
 def Bemenet(input_string):
     output = generator(os.environ['SECRET_PROMPT'] + " " + input_string, max_length=100)[0]['generated_text']
-    if os.environ['SECRET'] in output:
+    if "password".upper() in input_string.upper() or "secret".upper() in input_string.upper():
         return "Sorry, but I am unable to answer this question."
     else:
         return output