Spaces:
Runtime error
Runtime error
Update demo.py
Browse files
demo.py
CHANGED
@@ -41,7 +41,7 @@ def predict(image, question):
|
|
41 |
with tf.Session(config=sessionConfig) as sess:
|
42 |
sess.graph.finalize()
|
43 |
|
44 |
-
epoch =
|
45 |
emaSaver.restore(sess, config.weightsFile(epoch))
|
46 |
|
47 |
evalRes = model.runBatch(sess, data['data'], data['image'], False)
|
|
|
41 |
with tf.Session(config=sessionConfig) as sess:
|
42 |
sess.graph.finalize()
|
43 |
|
44 |
+
epoch = 25
|
45 |
emaSaver.restore(sess, config.weightsFile(epoch))
|
46 |
|
47 |
evalRes = model.runBatch(sess, data['data'], data['image'], False)
|