ALBORAQ-AI-IMMO / backend /test_api.py
mhdbbbbb's picture
Upload folder using huggingface_hub
fc41845 verified
import requests
import json
def test_prediction():
url = "http://localhost:8000/predict"
payload = {
"City": "Casablanca",
"Neighborhood": "Anfa",
"Type": "Appartement",
"Surface": 100.0,
"Rooms": 3,
"Bedrooms": 2,
"Standing": "Haut Standing",
"Residency": "Public / Quartier ouvert",
"Orientation": "Sud (Ensoleillé)",
"View": "Sans vis-à-vis",
"Condition": "Bon état",
"Floor": 2,
"Lift": 1,
"Pool": 0,
"Garden": 0,
"Parking_Spots": 1,
"Proximity_Tram": 1,
"Proximity_University": 0,
"Proximity_Mosque": 0
}
# Actually, matching the Pydantic model in main.py
payload = {
"City": "Casablanca",
"Neighborhood": "Anfa",
"Type": "Appartement",
"Surface": 100.0,
"Rooms": 3,
"Bedrooms": 2,
"Standing": "Haut Standing",
"Residency": "Public / Quartier ouvert",
"Orientation": "Sud (Ensoleillé)",
"View": "Sans vis-à-vis",
"Condition": "Bon état",
"Floor": 2,
"Lift": 1,
"Pool": 0,
"Garden": 0,
"Parking_Spots": 1,
"Proximity_Tram": 1,
"Proximity_University": 0,
"Proximity_Mosque": 0
}
try:
response = requests.post(url, json=payload)
print(f"Status: {response.status_code}")
print(f"Response: {json.dumps(response.json(), indent=2, ensure_ascii=False)}")
except Exception as e:
print(f"Connection error: {e}")
if __name__ == "__main__":
test_prediction()