File size: 1,123 Bytes
c2f0d2b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
import openai
import gradio as gr
openai.api_key = "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"
start_sequence = "\nAI:"
restart_sequence = "\nHuman: "
prompt = "请输入你的问题 "
def openai_create(prompt):
response = openai.Completion.create(
model="text-davinci-003",
prompt=prompt,
temperature=0.9,
max_tokens=3000,
top_p=1,
frequency_penalty=0,
presence_penalty=0.6,
stop=[" Human:", " AI:"]
)
return response.choices[0].text
def chatgpt_clone(input, history):
history = history or []
s = list(sum(history, ()))
s.append(input)
inp = ' '.join(s)
output = openai_create(inp)
history.append((input, output))
return history, history
block = gr.Interface(
fn=chatgpt_clone,
inputs=[
gr.inputs.Textbox(placeholder=prompt, label="问题"),
gr.inputs.Hidden(default=[]),
],
outputs=[
gr.outputs.Textbox(label="回答"),
gr.outputs.Hidden(),
],
title="清苑的AI",
theme="compact",
layout="vertical",
description=prompt,
allow_flagging=False,
)
block.launch()
|