Commit
•
769a633
1
Parent(s):
24e1943
Update app.py
Browse files
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/
|
25 |
words_dict=pickle.load(f)
|
26 |
|
27 |
|
28 |
-
vocab_size = 5611
|
29 |
-
|
30 |
|
31 |
-
MAX_LEN=192
|
32 |
-
|
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/
|
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 |
|