Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -37,7 +37,7 @@ def scrape(url: str = Query(...)):
|
|
| 37 |
return ThreadResponse(question="", replies=[])
|
| 38 |
|
| 39 |
|
| 40 |
-
MODEL_NAME = "
|
| 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)
|