Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import argparse
 import random
 
@@ -47,7 +48,7 @@ transform_image = transforms.Compose(
 )
 
 
-# @spaces.GPU()
+@spaces.GPU()
 def infer(
     prompt,
     image,