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