Update README.md

README.md

@@ -26,11 +26,7 @@ sys.path.insert(0, "bert_experimental")
 from bert_experimental.finetuning.text_preprocessing import build_preprocessor
 from bert_experimental.finetuning.graph_ops import load_graph
 
-
 df = pd.read_csv("test.tsv", sep='\t')
-# visual information   image caption
-# standard poodle shopping cart footwear 	a close up of a dog laying in a basket
-
 
 texts = []
 delimiter = " ||| "
@@ -38,12 +34,10 @@ delimiter = " ||| "
 for vis, cap  in zip(df.visual.tolist(), df.caption.tolist()):
   texts.append(delimiter.join((str(vis), str(cap))))
 
-
 texts = np.array(texts)
 
 trX, tsX = train_test_split(texts, shuffle=False, test_size=0.01)
 
-
 restored_graph = load_graph("frozen_graph.pb")
 
 graph_ops = restored_graph.get_operations()
@@ -59,16 +53,14 @@ py_func = tf.numpy_function(preprocessor, [x], [tf.int32, tf.int32, tf.int32], n
 py_func = tf.numpy_function(preprocessor, [x], [tf.int32, tf.int32, tf.int32])
 
 ##predictions
-
 sess = tf.Session(graph=restored_graph)
 
 print(trX[:4])
 
 y = tf.print(y, summarize=-1)
-#x = tf.print(x, summarize=-1)
 y_out = sess.run(y, feed_dict={
         x: trX[:4].reshape((-1,1))
- 	#x: trX[:90000].reshape((-1,1))
+ 	
     })