import gradio as gr
from transformers import pipeline

# Charger le modèle Bio-Medical-Llama-3-8B
model = pipeline("text-generation", model="ContactDoctor/Bio-Medical-Llama-3-8B")

# Définir la fonction de génération de texte
def generate_text(prompt):
    response = model(prompt, max_length=100)
    return response[0]["generated_text"]

# Créer l'interface Gradio
iface = gr.Interface(
    fn=generate_text,
    inputs="text",
    outputs="text",
    title="Medi-Trust API",
    description="Générez des réponses médicales à partir de déclarations."
)

# Lancer l'application
iface.launch()