pradelf commited on
Commit
75dfb95
·
verified ·
1 Parent(s): bbee45a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -3
app.py CHANGED
@@ -3,6 +3,13 @@ from pydantic import BaseModel
3
  import mlflow.pyfunc
4
  import pandas as pd
5
  import config
 
 
 
 
 
 
 
6
 
7
  # 1) URL du Space HF qui héberge MLflow (tracking server)
8
  SPACE_URL = "https://pradelf-getaround-api.hf.space" # <-- remplace ici
@@ -119,9 +126,10 @@ async def predict(predictionFeatures: RentalFeatures):
119
  # ]
120
 
121
  car_caracteristic = pd.DataFrame([predictionFeatures.model_dump()])
122
- print(car_caracteristic)
123
- #car_caracteristic = pd.DataFrame({"Car": pf})
124
 
 
 
125
  # Log model from mlflow
126
  logged_model = "models:/getaround-price-prediction-model/versions/2" # REPLACE WITH YOUR OWN RUN ID
127
 
@@ -134,5 +142,7 @@ async def predict(predictionFeatures: RentalFeatures):
134
  "prediction": prediction.tolist()[0],
135
  "detail": "Prédiction du tarif journalier (nul si aucun modèle : model.pkl).",
136
  }
137
- print(response)
 
 
138
  return response
 
3
  import mlflow.pyfunc
4
  import pandas as pd
5
  import config
6
+ import logging
7
+ import mylib
8
+ logger = logging.getLogger(__name__)
9
+
10
+
11
+ logging.basicConfig(filename='myapp.log', level=logging.INFO)
12
+
13
 
14
  # 1) URL du Space HF qui héberge MLflow (tracking server)
15
  SPACE_URL = "https://pradelf-getaround-api.hf.space" # <-- remplace ici
 
126
  # ]
127
 
128
  car_caracteristic = pd.DataFrame([predictionFeatures.model_dump()])
129
+ logger.info('Started')
 
130
 
131
+ logger.info(car_caracteristic)
132
+ #car_caracteristic = pd.DataFrame({"Car": pf})
133
  # Log model from mlflow
134
  logged_model = "models:/getaround-price-prediction-model/versions/2" # REPLACE WITH YOUR OWN RUN ID
135
 
 
142
  "prediction": prediction.tolist()[0],
143
  "detail": "Prédiction du tarif journalier (nul si aucun modèle : model.pkl).",
144
  }
145
+ logger.info(response)
146
+
147
+ logger.info('Finished')
148
  return response