Create app.py

app.py

@@ -0,0 +1,22 @@
+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()