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) | |