Update app.py

app.py

@@ -611,11 +611,11 @@ def main():
 
             model.compile(optimizer='adam', loss='mse')  # Compile the model
               # Explicitly build the model
-            plot_model(model, to_file='model_architecture.png', show_shapes=True, show_layer_names=True)
-            img = mpimg.imread('model_architecture.png')
-            plt.imshow(img)
-            plt.axis('off') 
-            plt.show()
+            # plot_model(model, to_file='model_architecture.png', show_shapes=True, show_layer_names=True)
+            # img = mpimg.imread('model_architecture.png')
+            # plt.imshow(img)
+            # plt.axis('off') 
+            # plt.show()
             model.build()
             return model
 
@@ -1151,11 +1151,11 @@ def main():
                 model.compile(optimizer='sgd', loss='mse') 
 
             model.build()  # Explicitly build the model
-            plot_model(model, to_file='model_architecture.png', show_shapes=True, show_layer_names=True)
-            img = mpimg.imread('model_architecture.png')
-            plt.imshow(img)
-            plt.axis('off') 
-            plt.show()
+            # plot_model(model, to_file='model_architecture.png', show_shapes=True, show_layer_names=True)
+            # img = mpimg.imread('model_architecture.png')
+            # plt.imshow(img)
+            # plt.axis('off') 
+            # plt.show()
             return model
 
         def train_regression_model(df):