Update app.py

app.py

@@ -19,7 +19,7 @@ import spaces  # Para ZeroGPU
 # =============================================================================
 
 class TextToMusicGenerator(nn.Module):
-    """Arquitetura do modelo Ricco"""
+    """Arquitetura do modelo Ricco (com discriminator para compatibilidade)"""
     
     def __init__(self, config):
         super().__init__()
@@ -69,6 +69,29 @@ class TextToMusicGenerator(nn.Module):
             nn.Tanh()
         )
         
+        # Discriminator (para compatibilidade com o modelo treinado)
+        self.discriminator = nn.Sequential(
+            nn.Conv2d(1, 32, 4, stride=2, padding=1),
+            nn.LeakyReLU(0.2),
+            
+            nn.Conv2d(32, 64, 4, stride=2, padding=1),
+            nn.BatchNorm2d(64),
+            nn.LeakyReLU(0.2),
+            
+            nn.Conv2d(64, 128, 4, stride=2, padding=1),
+            nn.BatchNorm2d(128),
+            nn.LeakyReLU(0.2),
+            
+            nn.Conv2d(128, 256, 4, stride=2, padding=1),
+            nn.BatchNorm2d(256),
+            nn.LeakyReLU(0.2),
+            
+            nn.AdaptiveAvgPool2d(1),
+            nn.Flatten(),
+            nn.Linear(256, 1),
+            nn.Sigmoid()
+        )
+        
         # Congelar BERT
         for param in self.text_encoder.parameters():
             param.requires_grad = False