Spaces:

SamiKLN
/

DocImageAI-Explorer

Sleeping

SamiKLN commited on Apr 27, 2025

Commit

9199f86

verified ·

1 Parent(s): 503425a

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -43,7 +43,7 @@ client = InferenceClient(token=HF_TOKEN)
 MODELS = {
     "summary": "facebook/bart-large-cnn",
     "caption": "Salesforce/blip-image-captioning-large",
-    "qa": "istilbert-base-cased-distilled-squad"  # plus léger
 }
 # Modèles Pydantic

 MODELS = {
     "summary": "facebook/bart-large-cnn",
     "caption": "Salesforce/blip-image-captioning-large",
+    "qa": "distilbert-base-cased-distilled-squad"  # plus léger
 }
 # Modèles Pydantic