File size: 581 Bytes
f446170
bd2b1d5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
f446170
 
 
bd2b1d5
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
import gradio as gr
from transformers import pipeline

# Load your fine-tuned model from Hugging Face Hub
model = pipeline("text-generation", model="your-huggingface-model-name")

def chatbot(input_text):
    # Generate text response
    response = model(input_text, max_length=100, do_sample=True)
    return response[0]["generated_text"]

# Create Gradio interface
interface = gr.Interface(
    fn=chatbot,
    inputs="text",
    outputs="text",
    title="LLM Chatbot",
    description="A chatbot powered by fine-tuned LLM."
)

if __name__ == "__main__":
    interface.launch()