import gradio as gr
from inference_fine_tune import generate_response

with gr.ChatInterface(
    fn=generate_response,
    title="🧠 Chat with Fine-Tuned Model",
    description="Ask me anything using your custom fine-tuned LLM.",
    theme="soft",
) as demo:
    demo.launch(share=True)