Krish-Upgrix commited on
Commit
b6eac4b
·
verified ·
1 Parent(s): 19c78f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ API_TOKEN = os.getenv("HF_API_TOKEN")
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 = "Qwen/Qwen2.5-Omni-7B" # 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}"}
 
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/codet5p-770m" # 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}"}