Update app.py

app.py

@@ -4,6 +4,10 @@ import matplotlib.pyplot as plt
 import numpy as np
 import os
 
+from datasets import load_dataset
+
+ds = load_dataset("OmarAmir2001/floor-plans-dataset")
+
 # Ensure directories exist
 os.makedirs("outputs/2d_drawings", exist_ok=True)
 os.makedirs("outputs/3d_models", exist_ok=True)