Llama2Chat / app.py
blacksmithop's picture
Update system prompt
69d661b
import gradio as gr
import random
import time
import requests
API_URL = "https://www.llama2.ai/api"
def send_request(question: str):
payload = {
"prompt": "[INST] QUESTION [/INST]".replace("QUESTION",question),
"version": "2796ee9483c3fd7aa2e171d38f4ca12251a30609463dcfd4cd76703f22e96cdf",
"systemPrompt": "You are a helpful assistant",
"temperature": 0.75,
"topP": 0.9,
"maxTokens": 800
}
r = requests.post(API_URL, json=payload)
if r.status_code == 200:
return r.content.decode("utf-8")
return "Sorry, I couldn't answer that question (network error)"
with gr.Blocks() as demo:
chatbot = gr.Chatbot()
msg = gr.Textbox()
clear = gr.ClearButton([msg, chatbot])
def respond(message, chat_history):
bot_message = send_request(question=message)
chat_history.append((message, bot_message))
time.sleep(2)
return "", chat_history
msg.submit(respond, [msg, chatbot], [msg, chatbot])
if __name__ == "__main__":
demo.launch()