ai_generate.py

import numpy as np
import pandas as pd
import torch
import openai

model = "gpt-3.5-turbo"
openai.api_key = os.environ['openai_key']

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