Update app.py

app.py

@@ -1,9 +1,13 @@
 import gradio as gr
-from transformers import AutoModelForCausalLM, AutoTokenizer
+from transformers import AutoTokenizer
+from peft import AutoPeftModelForCausalLM
 
 # Load the model and tokenizer from Hugging Face
 model_name = "PierreJousselin/lora_model"  # Replace with your model's name or path
-model = AutoModelForCausalLM.from_pretrained(model_name)
+model = AutoPeftModelForCausalLM.from_pretrained(
+        model_name,
+        
+    )
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
 # Define the chat function