File size: 1,281 Bytes
db6e2f8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 |
import os
import requests
import json
from io import BytesIO
from flask import Flask, jsonify, render_template, request, send_file
from modules.inference import infer_t5
from modules.dataset import query_emotion
# https://huggingface.co/settings/tokens
# https://huggingface.co/spaces/{username}/{space}/settings
API_TOKEN = os.getenv("BIG_GAN_TOKEN")
app = Flask(__name__)
@app.route("/")
def index():
return render_template("index.html")
@app.route("/infer_biggan")
def biggan():
input = request.args.get("input")
output = requests.request(
"POST",
"https://api-inference.huggingface.co/models/osanseviero/BigGAN-deep-128",
headers={"Authorization": f"Bearer {API_TOKEN}"},
data=json.dumps(input),
)
return send_file(BytesIO(output.content), mimetype="image/png")
@app.route("/infer_t5")
def t5():
input = request.args.get("input")
output = infer_t5(input)
return jsonify({"output": output})
@app.route("/query_emotion")
def emotion():
start = request.args.get("start")
end = request.args.get("end")
print(start)
print(end)
output = query_emotion(int(start), int(end))
return jsonify({"output": output})
if __name__ == "__main__":
app.run(host="0.0.0.0", port=7860)
|