Update app.py

app.py

@@ -7,9 +7,10 @@ import torch
 # model_name = "PierreJousselin/lora_model"  # Replace with the name you used on Hugging Face
 # model = AutoModelForCausalLM.from_pretrained(model_name, device_map="cpu")  # Force model to load on CPU
 
-model_id = "unsloth/Phi-3.5-mini-instruct"
-peft_model_id = "PierreJousselin/phi"
-
+#model_id = "unsloth/Phi-3.5-mini-instruct"
+#peft_model_id = "PierreJousselin/phi"
+model_id = "unsloth/Llama-3.2-1B-Instruct"
+peft_model_id = "PierreJousselin/llama"
 model = AutoModelForCausalLM.from_pretrained(model_id)
 model.load_adapter(peft_model_id)
 tokenizer = AutoTokenizer.from_pretrained(model_id)