| import gradio as gr | |
| from transformers import pipeline | |
| # Model load (stable + public) | |
| generator = pipeline( | |
| "text-generation", | |
| model="distilgpt2" | |
| ) | |
| # Chat function | |
| def chat(message): | |
| result = generator( | |
| message, | |
| max_new_tokens=100, | |
| do_sample=True, | |
| temperature=0.7 | |
| ) | |
| return result[0]["generated_text"] | |
| # Gradio interface (UI minimal, API usable) | |
| demo = gr.Interface( | |
| fn=chat, | |
| inputs=gr.Textbox(placeholder="Type..."), | |
| outputs=gr.Textbox(), | |
| title="ZPRIA AI" | |
| ) | |
| demo.launch() |