Roxie225 commited on
Commit
c29d60f
·
verified ·
1 Parent(s): 871639c

Create streamlit_app.py

Browse files
Files changed (1) hide show
  1. 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"])