ChickenChat / app.py
puffy310's picture
Create app.py
7e6ed4a verified
raw
history blame
No virus
845 Bytes
import gradio as gr
from random import randint
import os
from openai import OpenAI
tok = os.getenv('deepseekapi')
client = OpenAI(api_key=tok, base_url="https://api.deepseek.com")
def response(message, history):
if message=="How do I buy a yacht?":
response="You don't"
elif message=="你会说中文吗":
response="我是一只鸡"
else:
response = client.chat.completions.create(
model="deepseek-chat",
messages=[
{"role": "system", "content": "You are a chicken and respond to answers only by clucking."},
{"role": "user", "content": message},
],
max_tokens=144,
temperature=0.7,
stream=False)
response = response.choices[0].message.content
return response
gr.ChatInterface(response).launch()