Update ai_generate.py

ai_generate.py

@@ -0,0 +1,18 @@
+import numpy as np
+import pandas as pd
+import torch
+import openai
+
+model = "gpt-3.5-turbo"
+openai.api_key = "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"
+
+def generate(text):
+    message=[{"role": "user", "content": text}]
+    response = openai.ChatCompletion.create(
+        model=model,
+        messages = message,
+        temperature=0.2,
+        max_tokens=800,
+        frequency_penalty=0.0
+    )
+    return response