Update app.py

app.py

@@ -84,9 +84,16 @@ async def lifespan(app: FastAPI):
             repo_id=MODEL_REPO_ID,
             filename=MODEL_FILE_NAME
         )
-        
+    
         model = joblib.load(model_path)
         print(f"--- Pomyślnie pobrano i wczytano model z Huba: {MODEL_REPO_ID} ---")
+
+        # 🧹 Naprawa nazw kolumn – usuwamy spacje z przodu i końca
+        if hasattr(model, "feature_names_in_"):
+            clean_names = [f.strip() for f in model.feature_names_in_]
+            model.feature_names_in_ = clean_names
+            print("🧹 Oczyszczone feature_names_in_:", model.feature_names_in_)
+        print(f"--- Pomyślnie pobrano i wczytano model z Huba: {MODEL_REPO_ID} ---")
         print("Feature names in model:", model.feature_names_in_)
         
     except Exception as e: