Update app.py

app.py

@@ -35,12 +35,14 @@ def generate(slider_x, slider_y, prompt,
   
     return gr.update(label=comma_concepts_x, interactive=True),gr.update(label=comma_concepts_y, interactive=True), x_concept_1, x_concept_2, y_concept_1, y_concept_2, avg_diff_x_1, avg_diff_x_2, avg_diff_y_1, avg_diff_y_2, image
 
+@spaces.GPU
 def update_x(x,y,prompt, avg_diff_x_1, avg_diff_x_2, avg_diff_y_1, avg_diff_y_2):
     avg_diff = [avg_diff_x_1.cuda(), avg_diff_x_2.cuda()] 
     avg_diff_2nd = [avg_diff_y_1.cuda(), avg_diff_y_2.cuda()]
     image = clip_slider.generate(prompt, scale=x, scale_2nd=y, num_inference_steps=8, avg_diff=avg_diff,avg_diff_2nd=avg_diff_2nd) 
     return image
 
+@spaces.GPU
 def update_y(x,y,prompt, avg_diff_x_1, avg_diff_x_2, avg_diff_y_1, avg_diff_y_2):
     avg_diff = [avg_diff_x_1.cuda(), avg_diff_x_2.cuda()] 
     avg_diff_2nd = [avg_diff_y_1.cuda(), avg_diff_y_2.cuda()]