SUPER_CB / app.py
DevashishBhake's picture
Update app.py
5f5d2c7
import g4f
import gradio as gr
system_msg = """I want you to act as a machine learning engineer. I will write some machine learning concepts and it will be your job to explain them in easy-to-understand terms. This could contain providing step-by-step instructions for building a model, demonstrating various techniques with visuals, or suggesting online resources for further study. My first suggestion request is 'I have a dataset without labels. Which machine learning algorithm should I use?'"""
# system_msg = """"""
def predict(message, history):
history_openai_format = []
history_openai_format.append({"role": "user", "content": system_msg})
for human, assistant in history:
history_openai_format.append({"role": "user", "content": human })
history_openai_format.append({"role": "Bing", "content":assistant})
history_openai_format.append({"role": "user", "content": message})
response = g4f.ChatCompletion.create(
model="gpt-4",
provider=g4f.Provider.Bing,
messages=history_openai_format,
stream=True)
output = ""
for message in response:
output = output + message
yield output
response.close()
gr.ChatInterface(predict).queue().launch(share=False)