Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,13 +8,13 @@ import requests
|
|
| 8 |
import json
|
| 9 |
app = flask.Flask(__name__, template_folder="static")
|
| 10 |
def text_generate(prompt):
|
| 11 |
-
|
| 12 |
-
|
| 13 |
-
|
| 14 |
-
|
| 15 |
-
|
| 16 |
-
|
| 17 |
-
|
| 18 |
response = requests.post(url, json=data)
|
| 19 |
if response.status_code == 200:
|
| 20 |
data_str=response.json()
|
|
@@ -25,9 +25,9 @@ def text_generate(prompt):
|
|
| 25 |
slack_msg = slack_msg+data_str["completions"][i]["log_prob"]+"\n\n"
|
| 26 |
else:
|
| 27 |
slack_msg="Failed to send JSON message to URL"+response.status_code
|
| 28 |
-
|
| 29 |
-
|
| 30 |
-
|
| 31 |
@app.route("/")
|
| 32 |
def index():
|
| 33 |
return 'Hello'
|
|
|
|
| 8 |
import json
|
| 9 |
app = flask.Flask(__name__, template_folder="static")
|
| 10 |
def text_generate(prompt):
|
| 11 |
+
url = "http://34.122.217.42/complete_batch"
|
| 12 |
+
data = {
|
| 13 |
+
"debug_ext_path": "foo",
|
| 14 |
+
"context":prompt,
|
| 15 |
+
"top_p": 0.9,
|
| 16 |
+
"temp": 0.75}
|
| 17 |
+
try:
|
| 18 |
response = requests.post(url, json=data)
|
| 19 |
if response.status_code == 200:
|
| 20 |
data_str=response.json()
|
|
|
|
| 25 |
slack_msg = slack_msg+data_str["completions"][i]["log_prob"]+"\n\n"
|
| 26 |
else:
|
| 27 |
slack_msg="Failed to send JSON message to URL"+response.status_code
|
| 28 |
+
except:
|
| 29 |
+
slack_msg="Failed to send JSON message to URL"+response.status_code
|
| 30 |
+
return slack_msg
|
| 31 |
@app.route("/")
|
| 32 |
def index():
|
| 33 |
return 'Hello'
|