Update app.py

app.py

@@ -8,20 +8,14 @@ from huggingface_hub import login
 token = os.environ.get("HUGGING_FACE_HUB_TOKEN")
 login(token)
 
-base_model = "mistralai/Mistral-7B-Instruct-v0.1"
-
-tokenizer = AutoTokenizer.from_pretrained(base_model, token=True)
-model = AutoModelForCausalLM.from_pretrained(base_model, torch_dtype="auto", token=True)
 
 # ---------- STEP 1: Fine-tuned 모델 Git에서 clone ----------
 repo_url = "DMID23/MachineToolAgent"
 
 
 # ---------- STEP 2: Tokenizer와 모델 로드 ----------
-base_model = "mistralai/Mistral-7B-Instruct-v0.1"
-
+base_model = "mistralai/Mistral-7B-v0.1"
 tokenizer = AutoTokenizer.from_pretrained(base_model)
-
 model = AutoModelForCausalLM.from_pretrained(
     repo_url,
     torch_dtype=torch.float32,