Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
@@ -1,9 +1,9 @@
|
|
1 |
import numpy # for miscellaneous
|
2 |
-
import keras # for network
|
3 |
import gradio # for interface
|
4 |
import matplotlib.pyplot # for colormap
|
5 |
import matplotlib.colors # for color conversion
|
6 |
-
import huggingface_hub
|
7 |
|
8 |
# Loadin the model
|
9 |
model1 = huggingface_hub.from_pretrained_keras("cmudrc/microstructure-colorization")
|
|
|
1 |
import numpy # for miscellaneous
|
2 |
+
# import keras # for network
|
3 |
import gradio # for interface
|
4 |
import matplotlib.pyplot # for colormap
|
5 |
import matplotlib.colors # for color conversion
|
6 |
+
import huggingface_hub # for loading model
|
7 |
|
8 |
# Loadin the model
|
9 |
model1 = huggingface_hub.from_pretrained_keras("cmudrc/microstructure-colorization")
|