DnDItem / app.py
stale2000's picture
Update app.py
b898420
raw
history blame
803 Bytes
#!/usr/bin/env python
# coding: utf-8
import os
import openai
import gradio
openai.api_key = os.getenv('openaikey')
def predict(input, manual_query_repacement, history=[]):
if manual_query_repacement != "":
input = manual_query_repacement
response = openai.Completion.create(
model="text-davinci-003",
prompt=input,
temperature=0.9,
max_tokens=150,
top_p=1,
frequency_penalty=0,
presence_penalty=0.6)
# tokenize the new input sentence
responseText = response["choices"][0]["text"]
history.append((input, responseText))
return history, history
inputText= gradio.Textbox(value="tmp")
gradio.Interface(fn=predict,
inputs=[inputText,"text",'state'],
outputs=["chatbot",'state']).launch()