Update app.py

app.py

@@ -5,7 +5,8 @@ import numpy as np
 from itertools import product
 import torch.nn as nn
 import matplotlib.pyplot as plt
-import matplotlib.colors as mcolors LinearSegmentedColormap
+import matplotlib.colors as mcolors 
+from matplotlib.colors import LinearSegmentedColormap
 import io 
 from io import BytesIO # Import io then BytesIO 
 from PIL import Image, ImageDraw, ImageFont