Vishwas1 commited on
Commit
46861ae
1 Parent(s): 40b6644

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ import json
9
  app = flask.Flask(__name__, template_folder="static")
10
 
11
  #API_URL = "http://34.122.217.42/complete_batch"
12
- API_URL = "https://api-inference.huggingface.co/models/Salesforce/codegen-350M-multi"
13
  API_TOKEN = os.environ["API_TOKEN"]
14
  headers = {"Authorization": f"Bearer {API_TOKEN}"}
15
  def text_generate(prompt):
 
9
  app = flask.Flask(__name__, template_folder="static")
10
 
11
  #API_URL = "http://34.122.217.42/complete_batch"
12
+ API_URL = "https://api-inference.huggingface.co/models/Salesforce/codegen-16B-multi"
13
  API_TOKEN = os.environ["API_TOKEN"]
14
  headers = {"Authorization": f"Bearer {API_TOKEN}"}
15
  def text_generate(prompt):