add cache examples

app.py

@@ -68,11 +68,11 @@ class Tango:
             return list(self.chunks(outputs, samples))
 
 # Initialize TANGO
+if torch.cuda.is_available():
+    tango = Tango()
+else:
+    tango = Tango(device="cpu")
 
-tango = Tango(device="cpu")
-tango.vae.to("cuda")
-tango.stft.to("cuda")
-tango.model.to("cuda")
     
 @spaces.GPU(duration=60)
 def gradio_generate(prompt, steps, guidance):
@@ -136,7 +136,7 @@ gr_interface = gr.Interface(
         ["A helicopter is in flight"],
         ["A dog barking and a man talking and a racing car passes by"],
     ],
-    cache_examples=False, # Turn on to cache.
+    cache_examples=True, # Turn on to cache.
 )
 
 # Launch Gradio app