Update app.py

app.py

@@ -53,12 +53,16 @@ class SpriteGenerator(nn.Module):
             nn.BatchNorm2d(16),
             nn.ReLU(True),
             
-            # Layer finale modificato per corrispondere esattamente ai pesi
-            nn.ConvTranspose2d(16, 3, 4, 2, 1, bias=False),  # Output layer per RGB
-            nn.BatchNorm2d(3),  # Aggiunto BatchNorm
-            nn.ReLU(True),      # Aggiunto ReLU
+            # 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),
+            
+            nn.Conv2d(3, 3, 3, 1, 1, bias=True),  # Prima convoluzione di rifinitura
+            nn.BatchNorm2d(3),
+            nn.ReLU(True),
             
-            nn.Conv2d(3, 3, 3, 1, 1)  # Layer di output finale
+            nn.Conv2d(3, 3, 3, 1, 1, bias=True)  # Convoluzione finale
         )
         
         # Frame interpolator
@@ -98,6 +102,8 @@ class SpriteGenerator(nn.Module):
             # Generate frame
             frame_latent_reshaped = frame_latent.unsqueeze(2).unsqueeze(3)
             frame = self.generator(frame_latent_reshaped)
+            # Aggiungiamo Tanh finale per normalizzare l'output
+            frame = torch.tanh(frame)
             all_frames.append(frame)
         
         # Stack all frames