| import gradio as gr |
| from transformers import pipeline |
|
|
| |
| generator = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.2") |
|
|
| |
| def fruit_health_bot(message, history=[]): |
| prompt = f"You are FruitHealthGPT, a friendly expert on fruits and health benefits. Be concise and helpful.\nUser: {message}\nFruitHealthGPT:" |
| response = generator(prompt, max_new_tokens=100, do_sample=True, temperature=0.7) |
| answer = response[0]["generated_text"].split("FruitHealthGPT:")[-1].strip() |
| return answer |
|
|
| |
| iface = gr.ChatInterface( |
| fn=fruit_health_bot, |
| title="🍎 FruitHealthGPT", |
| description="Ask me about the health benefits of any fruit!" |
| ) |
|
|
| iface.launch() |
|
|
|
|
