specify instruct llama3

app.py

@@ -6,7 +6,7 @@ from datasets import load_dataset
 """
 For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
 """
-client = InferenceClient("meta-llama/Meta-Llama-3-8B")
+client = InferenceClient("meta-llama/Meta-Llama-3-8B-Instruct")
 
 
 def respond(
@@ -48,7 +48,7 @@ def train_model():
                            'train': 'training_set.json'})
 
     # Load model
-    model = AutoModelForCausalLM.from_pretrained('meta-llama/Meta-Llama-3-8B')
+    model = AutoModelForCausalLM.from_pretrained('meta-llama/Meta-Llama-3-8B-Instruct')
 
     # Define training arguments
     training_args = TrainingArguments(