Update app.py

app.py

@@ -1,5 +1,6 @@
 import os
 import numpy as np
+import spaces
 import torch
 import PIL.Image
 import gradio as gr
@@ -1035,6 +1036,7 @@ for network_name, network_info in networks.items():
     with dnnlib.util.open_url(network_info["path"]) as f:
         network_info["G"] = legacy.load_network_pkl(f)['G_ema'].to(device)
 
+@spaces.GPU
 def generate_random_images(network_choice, class_label=None, num_images=2):
     # Get the selected generator
     G = networks[network_choice]["G"]