File size: 786 Bytes
b7937ac 02892a4 b7937ac |
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 |
from flask import Flask, request, jsonify, send_from_directory
from SpaceGen.model import SpaceGenModel
from flask_cors import CORS
app = Flask(__name__, static_folder='.', static_url_path='')
CORS(app)
MODEL_PATH = "SpaceGen/SpaceGen_Large.keras"
model = SpaceGenModel(model_path=MODEL_PATH)
@app.route('/api/data', methods=['POST'])
def space_text():
if request.is_json:
data = request.get_json()
corrupted_text = data.get('corrupted_text')
spaced_text = model.fix_space(corrupted_text)
return jsonify({'spaced_text': spaced_text})
@app.get('/')
def index():
return send_from_directory('.', 'main.html')
import os
if __name__ == '__main__':
port = int(os.environ.get("PORT", 7860))
app.run(host="0.0.0.0", port=port, debug=False) |