chhetri123 commited on
Commit
769a633
1 Parent(s): 24e1943

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -21,15 +21,15 @@ from tensorflow.keras.applications.resnet import preprocess_input
21
 
22
 
23
  ResNet152Model=ResNet152(include_top=False, weights='imagenet',input_shape=(224,224,3), pooling='avg')
24
- with open("pickle_files/words_dict_40.pkl","rb") as f:
25
  words_dict=pickle.load(f)
26
 
27
 
28
- vocab_size = 5611
29
- # vocab_size=6031 # 40 epoch all data
30
 
31
- MAX_LEN=192
32
- # MAX_LEN=210 # 40 epoch all data
33
  inv_dict = {v:k for k, v in words_dict.items()}
34
 
35
 
@@ -52,7 +52,7 @@ outputs = Dense(vocab_size, activation='softmax')(decoder2)
52
  model = Model(inputs=[inputs1, inputs2], outputs=outputs)
53
  # compile model
54
  model.compile(loss='categorical_crossentropy', optimizer='adam',metrics=['accuracy'])
55
- model.load_weights("LSTM/resnet152_lstm_model_weights_45epoch.h5")
56
  #
57
  print("LSTM model loaded successfully")
58
 
 
21
 
22
 
23
  ResNet152Model=ResNet152(include_top=False, weights='imagenet',input_shape=(224,224,3), pooling='avg')
24
+ with open("pickle_files/words_dict_nepali_sc_40.pkl","rb") as f:
25
  words_dict=pickle.load(f)
26
 
27
 
28
+ # vocab_size = 5611
29
+ vocab_size=6031 # 40 epoch all data
30
 
31
+ # MAX_LEN=192
32
+ MAX_LEN=210 # 40 epoch all data
33
  inv_dict = {v:k for k, v in words_dict.items()}
34
 
35
 
 
52
  model = Model(inputs=[inputs1, inputs2], outputs=outputs)
53
  # compile model
54
  model.compile(loss='categorical_crossentropy', optimizer='adam',metrics=['accuracy'])
55
+ model.load_weights("LSTM/resnet152_lstm_model_weights_40epoch.h5")
56
  #
57
  print("LSTM model loaded successfully")
58