Update main.py

main.py

@@ -1,9 +1,9 @@
 import numpy # for miscellaneous
-import keras # for network
+# import keras # for network
 import gradio # for interface
 import matplotlib.pyplot # for colormap
 import matplotlib.colors # for color conversion
-import huggingface_hub
+import huggingface_hub # for loading model
 
 # Loadin the model
 model1 = huggingface_hub.from_pretrained_keras("cmudrc/microstructure-colorization")