Update main.py

main.py

@@ -5,11 +5,9 @@ import matplotlib.pyplot # for colormap
 import matplotlib.colors # for color conversion
 import huggingface_hub
 
+# Loadin the model
 model1 = huggingface_hub.from_pretrained_keras("cmudrc/microstructure-colorization")
 
-#Loading the weights in the architecture (The file should be stored in the same directory as the code)
-# model1.load_weights('modelV13_500trained_1.h5')
-
 # Get the color map by name:
 cm = matplotlib.pyplot.get_cmap('RdBu')