File size: 576 Bytes
ba0c334
eeeb58e
95da712
eeeb58e
95da712
 
 
 
eeeb58e
20c3fbb
95da712
c012653
20c3fbb
 
eeeb58e
20c3fbb
eeeb58e
20c3fbb
eeeb58e
 
53f41eb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
import os
import gradio as gr
from openai import OpenAI

client = OpenAI(
    base_url="https://router.huggingface.co/v1",
    api_key=os.getenv("HF_TOKEN")
)

def chat(message, history):
    response = client.chat.completions.create(
        model="mistralai/Mistral-7B-Instruct-v0.3",
        messages=[{"role": "user", "content": message}],
        max_tokens=256
    )
    return response.choices[0].message.content

demo = gr.ChatInterface(fn=chat, title="Simple Chat")

if __name__ == "__main__":
    demo.launch(server_name="0.0.0.0", server_port=7860, show_api=False)