Nikolay Ponomarev commited on
Commit
8bb366c
·
1 Parent(s): 9b111b7

Item Search

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ from transformers import pipeline, AutoTokenizer
12
  DEFAULT_INTENT_MODEL = os.getenv("INTENT_MODEL", "joeddav/xlm-roberta-large-xnli")
13
 
14
  # 2) Checklist generator (instruct)
15
- DEFAULT_GEN_MODEL = os.getenv("GEN_MODEL", "mistralai/Mistral-7B-Instruct-v0.2")
16
  # If your Space can handle it, this is often better structured:
17
  # DEFAULT_GEN_MODEL = os.getenv("GEN_MODEL", "Qwen/Qwen2.5-1.5B-Instruct")
18
 
 
12
  DEFAULT_INTENT_MODEL = os.getenv("INTENT_MODEL", "joeddav/xlm-roberta-large-xnli")
13
 
14
  # 2) Checklist generator (instruct)
15
+ DEFAULT_GEN_MODEL = os.getenv("GEN_MODEL", "Qwen/Qwen2.5-1.5B-Instruct")
16
  # If your Space can handle it, this is often better structured:
17
  # DEFAULT_GEN_MODEL = os.getenv("GEN_MODEL", "Qwen/Qwen2.5-1.5B-Instruct")
18