Update app.py

app.py

@@ -42,7 +42,6 @@ class VQVAE2(nn.Module):
 
 # Diffusion Model Definition
 class Diffusion(nn.Module):
-    # FIXED: Changed n_head default from 8 to 4 to make dimensions compatible
     def __init__(self, n_inputs=3, n_embed=512, n_head=4, n_layer=12):
         super().__init__()
         self.time_embed = nn.Embedding(1000, n_inputs * 4)
@@ -257,4 +256,5 @@ iface = gr.Interface(
 )
 
 # 🚀 Launch the Gradio interface
-iface.launch(enable_queue=True)
+# FIXED: Replaced deprecated 'enable_queue=True' with the .queue() method
+iface.queue().launch()