Update app.py

app.py

@@ -32,8 +32,9 @@ for j in range(6):
     norm_weights = scaler.fit_transform(weights)
     plt.subplot(2,3,j+1)
     plt.imshow(norm_weights,cmap='gray')
-    plt.title(f"Conv Layer")
+    #plt.title(f"Conv Layer")
+    plt.title(f"Filter {j}")
     plt.axis('off')
 plt.tight_layout()
-plt.show();
+st.pyplot(plt.gcf())