Update app.py

app.py

@@ -46,7 +46,6 @@ class ConditionalVAE(nn.Module):
         return self.decode(z, y), mu, logvar
 
 # Load model
-@st.cache_resource  # This will be ignored by Gradio but won't cause errors
 def load_model():
     model = ConditionalVAE(input_dim=784, hidden_dim=400, latent_dim=20, num_classes=10)
     model.load_state_dict(torch.load('mnist_cvae_model.pth', map_location=device))