Spaces:
Sleeping
Sleeping
import gradio as gr | |
import requests | |
import os | |
def function(Textbox,Textbox2): | |
target = os.environ.get("target") | |
target2 = os.environ.get("target2") | |
model = os.environ.get("model") | |
hrc = os.environ.get("hrc") | |
content = os.environ.get("content") | |
if Textbox2 == target: | |
payload = { | |
"model": "gpt-3.5-turbo", | |
"messages": [{"role": "system", "content": content},{"role": "user", "content": user}], | |
"temperature" : 1.0, | |
"top_p":1.0, | |
"n" : 1, | |
"stream": False, | |
"presence_penalty":0, | |
"frequency_penalty":0, | |
} | |
headers = { | |
"Content-Type": "application/json", | |
"Authorization": f"Bearer {target2}" | |
} | |
response = requests.post(hrc, headers=headers, json=payload, stream=False) | |
response = response.json() | |
return response["choices"][0]["message"]["content"] | |
else: | |
return "Failed" | |
inputs = [ | |
gr.inputs.Textbox(label="Textbox",type="text"), | |
gr.inputs.Textbox(label="Textbox2",type="password") | |
] | |
iface = gr.Interface(fn=function, inputs=inputs, outputs="text") | |
iface.launch() | |