Update app.py

app.py

@@ -3,9 +3,9 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 import gradio as gr
 
 # Adjust this to your model ID
-model_id = "decision-oaif/Meta-Llama-3-8B-Instruct-sft-intercode-bash-iter1"
+model_id = "decision-oaif/Meta-Llama-3.1-8B-Instruct-sft-intercode-bash-iter0"
 
-#peft_model_id = "decision-oaif/Meta-Llama-3-8B-Instruct-sft-intercode-bash-iter0"
+#peft_model_id = "decision-oaif/Meta-Llama-3.1-8B-Instruct-sft-intercode-bash-iter0"
 # Load model with device map and dtype
 model = AutoModelForCausalLM.from_pretrained(
     model_id,