clarkchan's picture
Update app.py
1967004
raw
history blame
546 Bytes
import gradio as gr
import requests
#from transformers import AutoTokenizer
#tokenizer = AutoTokenizer.from_pretrained("liam168/c2-roberta-base-finetuned-dianping-chinese")
def chat(input):
url = 'http://dark.21cnai.com:5000/api/chat'
data = { 'message': input }
headers = {'content-type': 'application/json','Authorization':'Bearer kdfjwoieskdflasdnf'}
response = requests.post(url, data=data, headers=headers)
return response
iface = gr.Interface(fn=chat, inputs="text", outputs=["text"])
iface.launch(server_name="0.0.0.0")