my_test_bot / app.py
Makaria's picture
Обновление кода с новым токеном
be0bcf5
raw
history blame
551 Bytes
import os
import gradio as gr
from huggingface_hub import InferenceClient
# Получаем токен из окружения
hf_token = os.getenv("HUGGINGFACE_TOKEN")
client = InferenceClient("sambanovasystems/SambaLingo-Russian-Chat", token=hf_token)
def respond(message):
response = client.chat_completion(messages=[{"role": "user", "content": message}])
return response['choices'][0]['message']['content']
# Интерфейс Gradio
iface = gr.Interface(fn=respond, inputs="text", outputs="text", title="AI Chat Bot")
iface.launch()