| import gradio as gr | |
| from transformers import pipeline | |
| generator = pipeline("text-generation", model="distilgpt2") | |
| def chat_fn(message, history=None): | |
| try: | |
| response = generator(message, max_length=100, pad_token_id=50256) | |
| bot_reply = response[0]['generated_text'].strip() | |
| return bot_reply, history or [] | |
| except Exception as e: | |
| return f"Error: {e}", history or [] | |
| gr.ChatInterface(fn=chat_fn).launch() |