Spaces:

ARCQUB
/

BPMN-entity-extractor

Running

ARCQUB commited on 1 day ago

Commit

3e49c61

verified ·

1 Parent(s): 9b79c93

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,8 +11,9 @@ MODEL_MAP = {
     #"Qwen": "models.qwen",
     #"Pixtral": "models.pixtral",
     #"Aya Vision": "models.aya_vision",
-    "GPT-4o": "models.gpt4o"
-    "GPT-4.1": "models.gpt_4.1"
 }
 # === Load Model

     #"Qwen": "models.qwen",
     #"Pixtral": "models.pixtral",
     #"Aya Vision": "models.aya_vision",
+    "GPT-4o": "models.gpt4o",
+    "GPT-4.1": "models.gpt_4.1",
+    "GPT-4.1-mini": "models.gpt_4.1_mini"
 }
 # === Load Model