Spaces:
Running
Running
| import gradio as gr | |
| from transformers import pipeline | |
| # Меньшая русская модель (работает на free tier) | |
| generator = pipeline("text-generation", model="sberbank-ai/rugpt3small_based_on_gpt2") | |
| def chat(message, history): | |
| # Промпт для стиля ChatGPT на русском | |
| prompt = f"Ты полезный ассистент. Отвечай кратко и точно. Вопрос: {message}\nОтвет:" | |
| response = generator(prompt, max_length=100, num_return_sequences=1, temperature=0.8)[0]['generated_text'] | |
| # Извлекаем ответ | |
| ai_reply = response.split("Ответ:")[-1].strip() | |
| history.append((message, ai_reply)) | |
| return history, "" | |
| # Интерфейс чата | |
| with gr.Blocks(title="Русский Чат-Бот") as demo: | |
| gr.Markdown("# Бесплатный ИИ-Чат на русском\nСпрашивай что угодно!") | |
| chatbot = gr.Chatbot(height=400) | |
| msg = gr.Textbox(placeholder="Напиши сообщение на русском...") | |
| clear = gr.Button("Очистить") | |
| msg.submit(chat, [msg, chatbot], [chatbot, msg]) | |
| clear.click(lambda: None, None, chatbot, queue=False) | |
| if __name__ == "__main__": | |
| demo.launch() |