vishnun commited on
Commit
c693d16
1 Parent(s): 081d5e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ import matplotlib.pyplot as plt
8
  inputs = gr.inputs.Image(shape=(256, 256))
9
  o1 = gr.outputs.Image()
10
  o2 = gr.outputs.Image()
11
- gen_model = load_model('model256.h5')
12
 
13
  def colorify(pixels):
14
 
 
8
  inputs = gr.inputs.Image(shape=(256, 256))
9
  o1 = gr.outputs.Image()
10
  o2 = gr.outputs.Image()
11
+ gen_model = load_model('256_model_250ep.h5')
12
 
13
  def colorify(pixels):
14