Krish-Upgrix commited on
Commit
e83ae8f
·
verified ·
1 Parent(s): 2b4371b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -7,9 +7,8 @@ API_TOKEN = os.getenv("HF_API_TOKEN")
7
 
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 = "Salesforce/CodeT5+" # 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}"}
 
7
 
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 = "bigcode/starcoder"
13
  API_URL = f"https://api-inference.huggingface.co/models/{MODEL_ID}"
14
  HEADERS = {"Authorization": f"Bearer {API_TOKEN}"}