Update app.py

Add spaces to utilize zeroGPU.

app.py

@@ -1,4 +1,5 @@
 import torch
+import spaces
 import gradio as gr
 import numpy as np
 import os
@@ -65,7 +66,8 @@ class DiffusionSampler:
         if self.vae is None:
             self.vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-ema").to(self.device)
             self.vae.eval()
-        
+
+    @spaces.GPU
     def generate_images(self, model, num_samples, genre, style, seed, progress=gr.Progress()):
         """Generate images with the DiT model"""
         global global_progress
@@ -159,6 +161,7 @@ def generate_random_seed():
     """Generate a random seed between 0 and 2^32 - 1"""
     return random.randint(0, 2**32 - 1)
 
+@spaces.GPU
 def generate_samples(num_samples, dit_size, genre_name, style_name, seed, progress=gr.Progress()):
     """Main function for Gradio interface"""
     if num_samples < 1 or num_samples > 16: