Spaces:
Sleeping
Sleeping
import gradio as gr | |
import openai | |
from openai import OpenAI | |
import os | |
def query(prompt): | |
client = client = OpenAI(api_key = os.getenv('OPENAPI_KEY')) # your own key | |
system_prompt = """ | |
You are a helpful text sentiment classifier. Answer ONLY using these three options: ['Positive', 'Negative', 'Neutral'] | |
""" | |
user_prompt = f""" | |
Classify the text into neutral, negative or positive. | |
Text: {prompt} | |
Sentiment: | |
""" | |
message = [{"role": "system", "content": f"{system_prompt}"},{"role": "user", "content": f"{user_prompt}"}] | |
response = client.chat.completions.create( | |
model="gpt-3.5-turbo-1106", | |
messages=message, | |
temperature=0.5 | |
) | |
return response.choices[0].message.content | |
gr.Interface(fn=query, inputs="textbox", outputs="textbox").launch() |