csuvikv commited on
Commit
be557e3
1 Parent(s): a586e97

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ generator = pipeline('text-generation', model='PygmalionAI/pygmalion-1.3b', retu
9
 
10
  def Bemenet(input_string):
11
  output = generator(os.environ['SECRET_PROMPT'] + " " + input_string, max_length=100)[0]['generated_text']
12
- if os.environ['SECRET'].upper() in output.upper():
13
  return "Sorry, but I am unable to answer this question."
14
  else:
15
  return output
 
9
 
10
  def Bemenet(input_string):
11
  output = generator(os.environ['SECRET_PROMPT'] + " " + input_string, max_length=100)[0]['generated_text']
12
+ if os.environ['SECRET'] in output:
13
  return "Sorry, but I am unable to answer this question."
14
  else:
15
  return output