Spaces:
Runtime error
Runtime error
File size: 911 Bytes
4404380 947d560 4404380 947d560 4404380 947d560 4404380 |
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 |
from flask import Flask, request, jsonify
import tensorflow as tf
import numpy as np
import os
app = Flask(__name__)
# Load the model
model_path = "sleep_cognition_model.h5"
if not os.path.exists(model_path):
raise FileNotFoundError(f"⚠️ Model file not found: {model_path}. Please check the path.")
model = tf.keras.models.load_model(model_path)
@app.route("/", methods=["GET"])
def home():
return jsonify({"message": "Sleep Cognition Model API is running!"})
@app.route("/predict", methods=["POST"])
def predict():
try:
data = request.json["input"]
data = np.array(data).reshape(1, -1) # Reshape input data if needed
prediction = model.predict(data)
return jsonify({"prediction": prediction.tolist()})
except Exception as e:
return jsonify({"error": str(e)}), 400
if __name__ == "__main__":
app.run(host="0.0.0.0", port=5000, debug=True)
|