yeah again 4

API_flask_server.py

@@ -1,37 +1,42 @@
 ## An python API for our Model
 ## Design by edander32 (Edmilson Alexandre) and jjambo(Joaquim Jambo)
 
-from flask import Flask, request, jsonify
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
 import pandas as pd
 import joblib
-
-app = Flask(__name__)
-
 from huggingface_hub import hf_hub_download
-import joblib
 
+# Inicializa o app
+app = FastAPI(title="Exoplanet Classifier API")
+
+# Baixa e carrega o modelo
 model_path = hf_hub_download(
-    repo_id="Edalexan/ia-pkl",  # substitui pelo teu nome/ID
+    repo_id="Edalexan/ia-pkl",  # substitua pelo seu repo/ID
     filename="exoplanet_model.pkl"
 )
 model = joblib.load(model_path)
 
+# Define o modelo de entrada
+class InputData(BaseModel):
+    # Substitua pelos nomes reais das features que seu modelo espera
+    feature1: float
+    feature2: float
+    feature3: float  # e assim por diante
 
-@app.route("/")
+@app.get("/")
 def home():
     return {"message": "API do classificador de exoplanetas online."}
 
-@app.route("/predict", methods=["POST"])
-def predict():
+@app.post("/predict")
+def predict(input_data: InputData):
     try:
-        data = request.get_json()
-        df = pd.DataFrame([data])
+        # Converte para DataFrame
+        df = pd.DataFrame([input_data.dict()])
         pred = model.predict(df)[0]
         prob = model.predict_proba(df)[0][1]
-        return jsonify({"prediction": int(pred), "probability": float(prob)})
+        return {"prediction": int(pred), "probability": float(prob)}
     except Exception as e:
-        return jsonify({"error": str(e)}), 400
+        raise HTTPException(status_code=400, detail=str(e))
 
-if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860)
 

Dockerfile

@@ -1,6 +1,6 @@
 FROM python:3.11-slim
 
-# Instalar dependências de sistema necessárias para LightGBM
+# Dependências de sistema para LightGBM
 RUN apt-get update && apt-get install -y \
     build-essential \
     libgomp1 \
@@ -9,12 +9,12 @@ RUN apt-get update && apt-get install -y \
 WORKDIR /app
 
 COPY requirements.txt .
-COPY API_flask_server.py .
-COPY model.py .
+COPY API_fastapi_server.py .
 
 RUN pip install --no-cache-dir -r requirements.txt
 
 EXPOSE 7860
 
-CMD ["python", "API_flask_server.py"]
+# Rodando FastAPI com uvicorn
+CMD ["uvicorn", "API_fastapi_server:app", "--host", "0.0.0.0", "--port", "7860"]
 

requirements.txt

@@ -4,4 +4,12 @@ pandas
 numpy
 scikit-learn
 lightgbm
+fastapi
+uvicorn
+pandas
+scikit-learn
+joblib
+huggingface_hub
+lightgbm
+