test17

main.py

@@ -1,9 +1,8 @@
 from flask import Flask, request
+from flask_cors import CORS, cross_origin
 import torch
 import model
 import numpy as np
-from sentence_transformers import SentenceTransformer
-sentence_model = SentenceTransformer("all-MiniLM-L6-v2")
 
 embedding_dim = 384
 hidden_dim = 512
@@ -16,6 +15,8 @@ lstm_model = model.LSTM(embedding_dim, hidden_dim, num_layers, output_dim)
 lstm_model.load_state_dict(torch.load('lstm.pt'))
 
 app = Flask(__name__)
+cors = CORS(app)
+app.config['CORS_HEADERS'] = 'Content-Type'
 
 def GeneratePosesJSON(input):
     with torch.no_grad():
@@ -37,10 +38,12 @@ def GeneratePosesJSON(input):
         return { 'people': data, 'animals': [], 'canvas_width': 900, 'canvas_height': 300 }
     
 @app.route('/')
+@cross_origin()
 def hello():
     return "Hello, World!"
 
 @app.route('/generate')
+@cross_origin()
 def generatePose():
     text = request.args.get('text')
     data = GeneratePosesJSON(text)

requirements.txt

@@ -1,4 +1,5 @@
 sentence-transformers
+flask-cors
 numpy
 torch
 Flask