Add application file6

app.py

@@ -31,8 +31,6 @@ ROOM_CATEGORIES = {
     'Storage': 10, 'Front Door': 11, 'Other': 13
 }
 
-rooms_data = []
-edges_data = []
 
 def save_samples(
         sample, ext, model_kwargs,
@@ -348,7 +346,8 @@ def create_layout(graphs, corners, room_type, metrics=False, use_ddim=True, ddim
     print(graphs, corners, room_type)
     model_kwargs = function_test(graphs, corners, room_type)
     for key in model_kwargs:
-        model_kwargs[key] = th.from_numpy(np.array([model_kwargs[key]])).cuda()
+        # model_kwargs[key] = th.from_numpy(np.array([model_kwargs[key]])).cuda()
+        model_kwargs[key] = th.from_numpy(np.array([model_kwargs[key]])).cpu()
 
     png_paths = []
     svg_paths = []
@@ -389,8 +388,11 @@ def create_layout(graphs, corners, room_type, metrics=False, use_ddim=True, ddim
     return png_paths, svg_paths
 
 
-def generate_layout(metrics: bool, ddim_steps: int, num_samples: int):
+rooms_data = []
+edges_data = []
+
 
+def generate_layout(metrics: bool, ddim_steps: int, num_samples: int):
     room_list = []
     room_corners = []
     living_room = 0