Update app.py

app.py

@@ -7,7 +7,7 @@ import gradio as gr
 # Define the VAE model
 class ConvVAE(nn.Module):
     def __init__(self, input_channels=3, latent_dim=32):
-        super(ImprovedConvVAE, self).__init__()
+        super(ConvVAE, self).__init__()
         self.latent_dim = latent_dim
         # Encoder
         self.enc_conv1 = nn.Conv2d(input_channels, 64, kernel_size=3, stride=2, padding=1)