Update app.py

app.py

@@ -4,6 +4,16 @@ import os
 import subprocess
 import torch
 
+from transformers import AutoModelForCausalLM, AutoTokenizer
+from huggingface_hub import login
+
+login("your_hf_token_here")
+
+base_model = "mistralai/Mistral-7B-Instruct-v0.1"
+
+tokenizer = AutoTokenizer.from_pretrained(base_model, use_auth_token=True)
+model = AutoModelForCausalLM.from_pretrained(base_model, torch_dtype="auto", use_auth_token=True)
+
 # ---------- STEP 1: Fine-tuned 모델 Git에서 clone ----------
 repo_url = "DMID23/MachineToolAgent"