nmarafo commited on
Commit
53b5731
·
verified ·
1 Parent(s): b11c580

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ print("CUDA version:", torch.version.cuda)
10
  # Load the Hugging Face token from secrets
11
  token = os.environ.get("HF_TOKEN")
12
 
13
- model_id = "meta-llama/Meta-Llama-3-8B-Instruct"
14
 
15
  # use quantization to lower GPU usage
16
  bnb_config = BitsAndBytesConfig(
 
10
  # Load the Hugging Face token from secrets
11
  token = os.environ.get("HF_TOKEN")
12
 
13
+ model_id = "google/shieldgemma-2b"
14
 
15
  # use quantization to lower GPU usage
16
  bnb_config = BitsAndBytesConfig(