Spaces:

Jofthomas
/

Magistral_geogussr_challenge

Paused

Jofthomas commited on May 26

Commit

000b2e5

verified ·

1 Parent(s): d446c6b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -79,7 +79,7 @@ GOOGLE_MAPS_API_KEY = os.getenv('GOOGLE_MAPS_API_KEY')
 SERVER_HF_TOKEN = os.getenv('HF_TOKEN', '')
-model_id = "mistralai/Mistral-Medium-3.5-128B"
 tokenizer = AutoTokenizer.from_pretrained(model_id, tokenizer_type="mistral", use_fast=False)
 model = Mistral3ForConditionalGeneration.from_pretrained(
         model_id, torch_dtype=torch.bfloat16, device_map="auto"

 SERVER_HF_TOKEN = os.getenv('HF_TOKEN', '')
+model_id = "mistralai/Magistral-Small-2509"
 tokenizer = AutoTokenizer.from_pretrained(model_id, tokenizer_type="mistral", use_fast=False)
 model = Mistral3ForConditionalGeneration.from_pretrained(
         model_id, torch_dtype=torch.bfloat16, device_map="auto"