Spaces:
Running
Running
Update app.py
Browse files
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-
|
| 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):
|