Create streamlit_app.py
Browse files- streamlit_app.py +29 -0
streamlit_app.py
ADDED
|
@@ -0,0 +1,29 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
|
| 2 |
+
import streamlit as st
|
| 3 |
+
from transformers import pipeline
|
| 4 |
+
|
| 5 |
+
st.set_page_config(page_title="Mini Q/A Médical avec LLM", layout="centered")
|
| 6 |
+
|
| 7 |
+
st.title("🧠 Assistant Médical IA (démo Hugging Face)")
|
| 8 |
+
st.write("Posez une question sur un texte médical et l'IA répondra.")
|
| 9 |
+
|
| 10 |
+
# Exemple de contexte médical simplifié
|
| 11 |
+
context = """
|
| 12 |
+
Mme Dupont, âgée de 58 ans, a été hospitalisée le 12 avril 2022 pour une bronchite chronique.
|
| 13 |
+
Elle a été traitée par antibiotiques (amoxicilline) pendant 7 jours, puis suivie en consultation
|
| 14 |
+
par le service de pneumologie.
|
| 15 |
+
"""
|
| 16 |
+
|
| 17 |
+
st.subheader("Contexte utilisé :")
|
| 18 |
+
st.code(context, language='markdown')
|
| 19 |
+
|
| 20 |
+
# Pipeline Hugging Face
|
| 21 |
+
qa_pipeline = pipeline("question-answering", model="distilbert-base-uncased-distilled-squad")
|
| 22 |
+
|
| 23 |
+
question = st.text_input("❓ Votre question", placeholder="Quels traitements Mme Dupont a-t-elle reçus ?")
|
| 24 |
+
|
| 25 |
+
if question:
|
| 26 |
+
with st.spinner("Recherche en cours..."):
|
| 27 |
+
result = qa_pipeline({"context": context, "question": question})
|
| 28 |
+
st.success("Réponse :")
|
| 29 |
+
st.write(result["answer"])
|