junaidbaber commited on
Commit
df369f2
·
verified ·
1 Parent(s): a7a70df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -3,7 +3,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
3
  import torch
4
 
5
  # Hugging Face repository details
6
- MODEL_ID = "meta-llama/CodeLlama-7b-Instruct-hf"
 
7
 
8
  def load_model():
9
  """Load the Hugging Face model and tokenizer."""
 
3
  import torch
4
 
5
  # Hugging Face repository details
6
+ # MODEL_ID = "meta-llama/CodeLlama-7b-Instruct-hf"
7
+ MODEL_ID = "meta-llama/Llama-3.1-8B-Instruct"
8
 
9
  def load_model():
10
  """Load the Hugging Face model and tokenizer."""