Update app.py

app.py

@@ -89,6 +89,7 @@ def get_normals(vertices: torch.Tensor, faces: torch.Tensor) -> torch.Tensor:
     return normals
 
 
+@spaces.GPU
 def generate(
     prompt: str,
     seed: int,
@@ -202,7 +203,6 @@ def launch_app(gen_fn: Callable):
     demo.queue().launch(share=False)
 
 
-@spaces.GPU
 def main():
     # Initialize the models and dataset
     diffuser, clip_model, dataset, device = init("config")