Update app.py

app.py

@@ -61,7 +61,7 @@ def plot_prediction(pred):
 # Define function to plot the decomposed prediction
 def plot_decomposed(decomposed):
     plt.figure(figsize=(4, 4))
-    plt.imshow(decomposed, origin="lower", norm=LogNorm())
+    plt.imshow(decomposed, origin="lower") #, norm=LogNorm())
 
     N = int(np.max(decomposed))
     for i in range(N):