Update app.py

app.py

@@ -1,10 +1,9 @@
-# app.py (API FastAPI)
 import os
 import tempfile
 from fastapi import FastAPI, UploadFile, File, HTTPException
 from fastapi.responses import JSONResponse
 import tensorflow as tf
-from prediction import predict_eeg_recording  # à adapter selon ton nom de module
+from prediction import predict_eeg_recording
 
 app = FastAPI(title="EEG Epilepsy Prediction API")
 model = tf.keras.models.load_model('model1_2dcnn.h5')
@@ -33,3 +32,7 @@ async def predict_endpoint(file: UploadFile = File(...)):
         "segment_probabilities": [float(p) for p in segment_probs]
     }
     return JSONResponse(content=response)
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run("app:app", host="0.0.0.0", port=7860, reload=True)