Upload 3 files
Browse files
src/hf_spaces_project/README.md
ADDED
|
@@ -0,0 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
|
| 2 |
+
# Assistant IA Médical - Mini Projet Hugging Face Spaces
|
| 3 |
+
|
| 4 |
+
Ce projet est une démonstration simple d'un assistant médical IA basé sur un modèle LLM (DistilBERT).
|
| 5 |
+
L'utilisateur pose une question, et l'IA cherche une réponse dans un texte médical.
|
| 6 |
+
|
| 7 |
+
## Fonctions :
|
| 8 |
+
- Question/réponse avec Hugging Face `transformers`
|
| 9 |
+
- Interface avec Streamlit
|
| 10 |
+
- Hébergeable sur Hugging Face Spaces
|
| 11 |
+
|
| 12 |
+
## Pour lancer localement :
|
| 13 |
+
```
|
| 14 |
+
pip install -r requirements.txt
|
| 15 |
+
streamlit run app.py
|
| 16 |
+
```
|
src/hf_spaces_project/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"])
|
src/hf_spaces_project/requirements.txt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
|
| 2 |
+
streamlit
|
| 3 |
+
transformers
|