diff --git a/program.py b/program.py
index c61dc3b..6335a1a 100644
--- a/program.py
+++ b/program.py
@@ -5,7 +5,7 @@ from sklearn import preprocessing
 import pandas as pd
 from gensim.test.utils import common_texts
 from gensim.models import Word2Vec
-
+00
 with open("train/in.tsv") as f:
     X_train = f.readlines()
     with open("train/expected.tsv") as ff:
@@ -13,10 +13,11 @@ with open("train/in.tsv") as f:
         with open("test-A/in.tsv") as d:
             X = d.readlines() 
 
-model = Word2Vec(X_train, min_count=1,size= 50,workers=3, window =3, sg = 1)
-X_train=model[X_train]
-X=model[X]
-FEAUTERES=1000
+model = Word2Vec(X_train, min_count=1,size= 500,workers=3, window =3, sg = 1)
+model1 = Word2Vec(X, min_count=1,size= 500,workers=3, window =3, sg = 1)
+X_train=model.wv
+X=model1.wv
+FEAUTERES=500
 
 class NeuralNetworkModel(torch.nn.Module):