Spaces:

CVPR
/

v-doc_abstractive_mac

Runtime error

ydin0771 commited on Jun 14, 2022

Commit

004447e

1 Parent(s): a482138

Update demo.py

Files changed (1) hide show

demo.py CHANGED Viewed

@@ -41,7 +41,7 @@ def predict(image, question):
   with tf.Session(config=sessionConfig) as sess:
     sess.graph.finalize()
-    epoch = loadWeights(sess, saver, init)
     emaSaver.restore(sess, config.weightsFile(epoch))
     evalRes = model.runBatch(sess, data['data'], data['image'], False)

   with tf.Session(config=sessionConfig) as sess:
     sess.graph.finalize()
+    epoch = 25
     emaSaver.restore(sess, config.weightsFile(epoch))
     evalRes = model.runBatch(sess, data['data'], data['image'], False)