Update app.py

app.py

@@ -53,7 +53,6 @@ class SpriteGenerator(nn.Module):
             nn.BatchNorm2d(16),
             nn.ReLU(True),
             
-            # Layers finali modificati per corrispondere esattamente ai pesi
             nn.ConvTranspose2d(16, 3, 4, 2, 1, bias=False),  # -> 3 x 512 x 512
             nn.BatchNorm2d(3),
             nn.ReLU(True),
@@ -62,6 +61,10 @@ class SpriteGenerator(nn.Module):
             nn.BatchNorm2d(3),
             nn.ReLU(True),
             
+            nn.Conv2d(3, 3, 3, 1, 1, bias=True),  # Seconda convoluzione di rifinitura
+            nn.BatchNorm2d(3),
+            nn.ReLU(True),
+            
             nn.Conv2d(3, 3, 3, 1, 1, bias=True)  # Convoluzione finale
         )