Update ai_generate.py

ai_generate.py

@@ -11,9 +11,11 @@ from transformers import pipeline
 def generate(text, model, api):
     if model == "GPT-Neo":
         response = pipeline("text-generation", model="EleutherAI/gpt-neo-2.7B")(text)
+        print(response)
         return response[0]
     elif model == "Llama 3":
         response = pipeline("text-generation", model="meta-llama/Meta-Llama-3-8B")(text)
+        print(response)
         return response[0] 
     elif model == "OpenAI GPT 3.5":
         client = OpenAI(