gouravbhadraDev commited on
Commit
2b9bbfd
·
verified ·
1 Parent(s): 6c0215b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ def scrape(url: str = Query(...)):
37
  return ThreadResponse(question="", replies=[])
38
 
39
 
40
- MODEL_NAME = "sarvamai/sarvam-m"
41
 
42
  # Load tokenizer and model once at startup, with device auto-mapping
43
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 
37
  return ThreadResponse(question="", replies=[])
38
 
39
 
40
+ MODEL_NAME = "mistralai/Magistral-Small-2506"
41
 
42
  # Load tokenizer and model once at startup, with device auto-mapping
43
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)