Add application file3

app.py

@@ -346,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 = []