Hugging Face's logo

Spaces:
Dama03
/
medical
Running

App Files Community
Dama03 commited on
Commit
d78c2da
·
1 Parent(s): a199ea0

jdsj

Browse files
Files changed (1) hide show
  1. medical_ai.py +1 -1
medical_ai.py CHANGED
@@ -22,7 +22,7 @@ logger = logging.getLogger(__name__)
22
  EMBEDDING_MODEL_NAME = "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2"
23
  NLLB_MODEL_NAME = "facebook/nllb-200-distilled-600M"
24
  # MODÈLE PRINCIPAL - MEDIUM pour la compétition
25
- MODEL_NAME = "microsoft/DialoGPT-medium"
26
  PATIENT_RECORDS_PATH = "patient_records.json"
27
 
28
  # Configuration optimisée pour CPU avec performance maximale
 
22
  EMBEDDING_MODEL_NAME = "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2"
23
  NLLB_MODEL_NAME = "facebook/nllb-200-distilled-600M"
24
  # MODÈLE PRINCIPAL - MEDIUM pour la compétition
25
+ MODEL_NAME = "google/flan-t5-base"
26
  PATIENT_RECORDS_PATH = "patient_records.json"
27
 
28
  # Configuration optimisée pour CPU avec performance maximale