Southampton-GPT / app.py
Slycat's picture
Update app.py
6480db9
raw
history blame
1.66 kB
from gradio_client import Client
from hugchat import hugchat
from hugchat.login import Login
import json
import gradio as gr
chat_client = Client("https://mosaicml-mpt-30b-chat.hf.space/")
retrieval = Client("https://slycat-southampton-similarity.hf.space/")
n_conv = 0
## Instruction: You are an AI language model and must return truthful responses as per the information below\n ##Input: Information: Your name is IITIGPT. You are a helpful and truthful chatbot. You can help answer any questions about the IIT Indore campus."
init_prompt =""
info="Information: \n"
q_prompt="\n ##Instruction: Please provide an appropriate response to the following: \n"
def change_conv():
# Create a new conversation
id = chatbot.new_conversation()
chatbot.change_conversation(id)
chatbot.chat(init_prompt)
chatbot.cookies = {}
def main(question):
global n_conv
# if(n_conv > 3):
# n_conv = 0
# change_conv(chatbot)
information = retrieval.predict(question, api_name = "/predict")
client = Client("https://mosaicml-mpt-30b-chat.hf.space/")
result = chat_client.predict(
"Howdy!", # str in 'System Prompt' Textbox component
"new.json",
"You are an AI language model and must return truthful responses as per the information below\n ##Input: Information: Your name is Southampton GPT. You are a helpful and truthful chatbot. You can help answer any questions about Southampton University."
+ information + question,
fn_index=3
)
n_conv+=1
print(result)
return result
demo = gr.Interface(main,"text","text")
if __name__ == "__main__":
demo.launch()