Update demo.py

demo.py

@@ -1,6 +1,6 @@
 
 import json
-
+import os
 import werkzeug
 import tensorflow as tf
 
@@ -13,8 +13,32 @@ import warnings
 
 def predict(image, question):
   parseArgs()
-  configPDF()
+  config.parallel = True
+  config.evalTrain = True
+  config.retainVal = True
+  config.useEMA = True
+  config.lrReduce = True
+  config.adam = True
+  config.clip = True
+  config.memoryVariationalDropout = True
+  config.relu='ELU'
+  config.encBi = True
+  config.wrdEmbRandom = True
+  config.wrdEmbUniform = True
+  config.outQuestion = True
+  config.initCtrl='Q'
+  config.controlContextual = True
+  config.controlInputUnshared = True
+  config.readProjInputs = True
+  config.readMemConcatKB = True
+  config.readMemConcatProj = True
+  config.readMemProj = True
+  config.readCtrl = True
+  config.writeMemProj = True
+  config.restore = True
   config.expName = 'PDF_exp_extra'
+  config.netLength = 16
+  configPDF()
   with open(config.configFile(), "a+") as outFile:
       json.dump(vars(config), outFile)
 
@@ -41,8 +65,10 @@ def predict(image, question):
   with tf.Session(config=sessionConfig) as sess:
     sess.graph.finalize()
 
-    epoch = 25
-    emaSaver.restore(sess, config.weightsFile(epoch))
+    # epoch = loadWeights(sess, saver, init)
+    print('###############', config.weightsFile(25))
+    os.system('ls -l ./weights/PDF_exp_extra')
+    emaSaver.restore(sess, config.weightsFile(25))
 
     evalRes = model.runBatch(sess, data['data'], data['image'], False)
     answer = None