saderat / app.py
Hadigoli's picture
Update app.py
0dd05fd verified
raw
history blame contribute delete
827 Bytes
import gradio as gr
from transformers import pipeline
# بارگذاری مدل چت بات
chatbot = pipeline("text-generation", model="OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5")
# تابع برای پاسخ به سوالات کاربر
def respond(message):
response = chatbot(message, max_length=100)[0]['generated_text']
return response
# ایجاد رابط کاربری با Gradio
iface = gr.Interface(
fn=respond, # تابع پاسخ‌دهی
inputs="text", # ورودی کاربر (متن)
outputs="text", # خروجی (پاسخ چت بات)
title="چت بات هوش مصنوعی", # عنوان رابط کاربری
description="یک چت بات ساده با استفاده از Hugging Face و Gradio." # توضیحات
)
# اجرای رابط کاربری
iface.launch()