# Importowanie potrzebnych bibliotek
import pickle
import pandas as pd

#Nazwa pliku z modelem
filename = "flat_model.pkl"

#Ładowanie modelu
model = pickle.load(open(filename, "rb"))

#Definicja funkcji predykcyjnej
def predict(stan, czynsz, liczba_pokoi, metraz, rynek):
  params = pd.DataFrame.from_dict({'stan': [stan], 'czynsz': [czynsz], 'liczba pokoi': [liczba_pokoi], 'metraz': [metraz], 'rynek': [rynek]})
  return model.predict(params).item()