Update ai_generate.py

ai_generate.py

@@ -9,13 +9,13 @@ pipes = {
 }
 
 def generate(text, model, api):
-    if model is "GPT-Neo":
+    if model == "GPT-Neo":
         response = pipes[model](text)
         return response[0]
-    elif model is "Llama 3":
+    elif model == "Llama 3":
         response = pipes[model](text)
         return response[0] 
-    elif model is "OpenAI GPT 3.5":
+    elif model == "OpenAI GPT 3.5":
         client = OpenAI(
             api_key=api,
         )
@@ -28,7 +28,7 @@ def generate(text, model, api):
             frequency_penalty=0.0
         )
         return response[0].message.content
-    elif model is "OpenAI GPT 4":
+    elif model == "OpenAI GPT 4":
         client = OpenAI(
             api_key=api,
         )
@@ -41,7 +41,7 @@ def generate(text, model, api):
             frequency_penalty=0.0
         )
         return response[0].message.content
-    elif model is "OpenAI GPT 4o":
+    elif model == "OpenAI GPT 4o":
         client = OpenAI(
             api_key=api,
         )