Krish-Upgrix commited on
Commit
66a0f7a
·
verified ·
1 Parent(s): d3edf19

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,8 +8,8 @@ API_TOKEN = os.getenv("HF_API_TOKEN")
8
 
9
  # Change MODEL_ID to a better model
10
  # MODEL_ID = "Salesforce/codet5p-770m" # CodeT5+ (Recommended)
11
- MODEL_ID = "bigcode/starcoder2-15b" # StarCoder2
12
- # MODEL_ID = "meta-llama/CodeLlama-34b-Instruct" # Code Llama
13
  # MODEL_ID = "bigcode/starcoder"
14
  API_URL = f"https://api-inference.huggingface.co/models/{MODEL_ID}"
15
  HEADERS = {"Authorization": f"Bearer {API_TOKEN}"}
 
8
 
9
  # Change MODEL_ID to a better model
10
  # MODEL_ID = "Salesforce/codet5p-770m" # CodeT5+ (Recommended)
11
+ # MODEL_ID = "bigcode/starcoder2-15b" # StarCoder2
12
+ MODEL_ID = "meta-llama/CodeLlama-34b-Instruct" # Code Llama
13
  # MODEL_ID = "bigcode/starcoder"
14
  API_URL = f"https://api-inference.huggingface.co/models/{MODEL_ID}"
15
  HEADERS = {"Authorization": f"Bearer {API_TOKEN}"}