data-ject commited on
Commit
11e7fea
·
verified ·
1 Parent(s): 4d1bd3a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ if not api_token:
15
  raise ValueError("HUGGINGFACE_API_TOKEN environment variable is not set")
16
 
17
  # Load model and tokenizer from Hugging Face Model Hub with authentication
18
- model_name = "meta-llama/Meta-Llama-3-70B-Instruct"
19
  tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=api_token)
20
  model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=api_token)
21
 
 
15
  raise ValueError("HUGGINGFACE_API_TOKEN environment variable is not set")
16
 
17
  # Load model and tokenizer from Hugging Face Model Hub with authentication
18
+ model_name = "meta-llama/Meta-Llama-3-8B"
19
  tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=api_token)
20
  model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=api_token)
21