makekali commited on
Commit
e2ae74a
·
verified ·
1 Parent(s): 966d889

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ HF_API_TOKEN = os.getenv("HF_TOKEN")
8
  if not HF_API_TOKEN:
9
  raise ValueError("HF_TOKEN environment variable is not set. Please set it to your Hugging Face API token.")
10
 
11
- MODEL_ID = "bigcode/starcoderbase-1b"
12
  API_URL = f"https://api-inference.huggingface.co/models/{MODEL_ID}"
13
  HEADERS = {"Authorization": f"Bearer {HF_API_TOKEN}"}
14
 
 
8
  if not HF_API_TOKEN:
9
  raise ValueError("HF_TOKEN environment variable is not set. Please set it to your Hugging Face API token.")
10
 
11
+ MODEL_ID = "google/flan-t5-base"
12
  API_URL = f"https://api-inference.huggingface.co/models/{MODEL_ID}"
13
  HEADERS = {"Authorization": f"Bearer {HF_API_TOKEN}"}
14