Update app.py

app.py

@@ -26,10 +26,10 @@ with st.sidebar:
 
 # --- Load Model & Tokenizer ---
 @st.cache_resource
-def load_model():  # amiguel/GM_Mistral7B_Finetune amiguel/GM_Qwen1.8B_Finetune
+def load_model():  #  amiguel/GM_Qwen1.8B_Finetune
     tokenizer = AutoTokenizer.from_pretrained("amiguel/GM_Mistral7B_Finetune", trust_remote_code=True, token=HF_TOKEN)
     model = AutoModelForCausalLM.from_pretrained(
-        "amiguel/GM_Mistral7B_FP16",
+        "amiguel/GM_Mistral7B_Finetune",
         device_map="auto",
         torch_dtype=torch.bfloat16 if torch.cuda.is_bf16_supported() else torch.float32,
         trust_remote_code=True,