Update ai_generate.py

ai_generate.py

@@ -3,17 +3,17 @@ from openai import OpenAI
 import os
 from transformers import pipeline
 
-pipes = {
-    'GPT-Neo': pipeline("text-generation", model="EleutherAI/gpt-neo-2.7B"),
-    'Llama 3': pipeline("text-generation", model="meta-llama/Meta-Llama-3-8B")
-}
+# pipes = {
+#     'GPT-Neo': pipeline("text-generation", model="EleutherAI/gpt-neo-2.7B"),
+#     'Llama 3': pipeline("text-generation", model="meta-llama/Meta-Llama-3-8B")
+# }
 
 def generate(text, model, api):
     if model == "GPT-Neo":
-        response = pipes[model](text)
+        response = pipeline("text-generation", model="EleutherAI/gpt-neo-2.7B")(text)
         return response[0]
     elif model == "Llama 3":
-        response = pipes[model](text)
+        response = pipeline("text-generation", model="meta-llama/Meta-Llama-3-8B")(text)
         return response[0] 
     elif model == "OpenAI GPT 3.5":
         client = OpenAI(