ccm commited on
Commit
e765410
1 Parent(s): 431046d

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -28,7 +28,7 @@ import matplotlib.colors # for color conversion
28
 
29
  from huggingface_hub import from_pretrained_keras
30
 
31
- model = from_pretrained_keras("cmudrc/microstructure-colorization")
32
 
33
 
34
  #Loading the weights in the architecture (The file should be stored in the same directory as the code)
 
28
 
29
  from huggingface_hub import from_pretrained_keras
30
 
31
+ model1 = from_pretrained_keras("cmudrc/microstructure-colorization")
32
 
33
 
34
  #Loading the weights in the architecture (The file should be stored in the same directory as the code)