File size: 475 Bytes
8ad8fd9 | 1 2 3 4 5 6 7 8 9 10 11 12 | # Store API keys in HF Space Settings -> Variables & Secrets
OPENAI_KEY = os.environ.get("OPENAI_KEY")
def call_gpt(prompt: str):
import requests
headers = {"Authorization": f"Bearer {OPENAI_KEY}"}
response = requests.post(
"https://api.openai.com/v1/chat/completions",
json={"model": "gpt-3.5-turbo", "messages": [{"role": "user", "content": prompt}]},
headers=headers
)
return response.json()["choices"][0]["message"]["content"] |