Update to account for MobileLLM-Pro repository split

app.py

@@ -9,7 +9,6 @@ from huggingface_hub import login
 import spaces
 
 MODEL_ID = "facebook/MobileLLM-Pro"
-SUBFOLDER = "instruct"           # use the chat template
 MAX_NEW_TOKENS = 256
 TEMPERATURE = 0.7
 TOP_P = 0.95
@@ -32,12 +31,11 @@ def _ensure_loaded():
     if _tokenizer is not None and _model is not None:
         return
     _tokenizer = AutoTokenizer.from_pretrained(
-        MODEL_ID, trust_remote_code=True, subfolder=SUBFOLDER
+        MODEL_ID, trust_remote_code=True
     )
     _model = AutoModelForCausalLM.from_pretrained(
         MODEL_ID,
         trust_remote_code=True,
-        subfolder=SUBFOLDER,
         torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
         low_cpu_mem_usage=True,
         device_map="auto" if torch.cuda.is_available() else None,