Update app.py

app.py

@@ -12,41 +12,37 @@ def classify_gender_equality(password, input_sentence):
 
     # set up OpenAI API key
     openai.api_key = password
+    
     # Set up the fine-tuned model for gender equality classification
-    model_engine = "text-davinci-003"
+    # model_engine = "text-davinci-003"
     model_prompt = ("""
     Please classify the following sentence as promoting or not promoting gender equality:
 
     Sentence: 
     """)
 
-    # The gender equality classification API endpoint
-    endpoint = "text-davinci-003"
-
     def is_gender_equal(sentence):
-        prompt = model_prompt + sentence
-        completions = openai.Completion.create(
-            engine=endpoint,
-            prompt=prompt,
-            max_tokens=1024,
-            n=1,
-            stop=None,
-            temperature=0.5,
-        )
-        message = completions.choices[0].text
-        return message.strip().lower() == "promoting gender equality"
+      prompt = model_prompt + sentence
+      completions = openai.Completion.create(
+          engine="text-davinci-003", # The gender equality classification API endpoint
+          prompt=prompt,
+          max_tokens=1024,
+          n=1,
+          stop=None,
+          temperature=0.5,
+      )
+      message = completions.choices[0].text
+      return message.strip().lower()
+
+  
+
 
     # Example usage
     # sentence = "Women are capable as men in leadership roles."
     # sentence = "Women are not as capable as men in leadership roles."
     # sentence = "Gender equality is important for the progress of society."
-    sentence = input_sentence
-    if is_gender_equal(sentence):
-        # print(f"'{sentence}' is promoting gender equality.")
-        return "This sentence is promoting gender equality."
-    else:
-        # print(f"'{sentence}' is not promoting gender equality.")
-        return "This sentence is not promoting gender equality."
+    
+    return "This Sentence is " + is_gender_equal(input_sentence)
 
 # Create the input text fields
 password_input = gr.inputs.Textbox(label="OpenAI API key", type="password")