Update app.py

app.py

@@ -238,17 +238,17 @@ class Tango:
 # Initialize TANGO
 
 tango = Tango(device="cpu")
-    
+tango.vae.to(device_type)
+tango.stft.to(device_type)
+tango.model.to(device_type)
+
 pipe = Tango2Pipeline(vae=tango.vae,
                       text_encoder=tango.model.text_encoder,
                       tokenizer=tango.model.tokenizer,
                       unet=tango.model.unet,
                       scheduler=tango.scheduler
                       )
-pipe.to(device_type)
-#tango.vae.to(device_type)
-#tango.stft.to(device_type)
-#tango.model.to(device_type)
+
     
 @spaces.GPU(duration=60)
 def gradio_generate(prompt, output_format, steps, guidance):