|
|
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) |
