Ahmed-El-Sharkawy commited on
Commit
4a02094
1 Parent(s): 2b51d5e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -68,7 +68,7 @@ preprocess = transforms.Compose([
68
 
69
  # Load Meta's LLaMA model for generating product descriptions
70
  def load_llama():
71
- model_name = "meta-llama/Meta-Llama-3-8B-Instruct"
72
  token = os.getenv("HUGGINGFACE_TOKEN")
73
  tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=token)
74
  model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=token).to(device)
 
68
 
69
  # Load Meta's LLaMA model for generating product descriptions
70
  def load_llama():
71
+ model_name = "meta-llama/Llama-3.2-1B-Instruct"
72
  token = os.getenv("HUGGINGFACE_TOKEN")
73
  tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=token)
74
  model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=token).to(device)