billusanda007 commited on
Commit
bcbc731
·
verified ·
1 Parent(s): 00f03a4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -63,7 +63,8 @@ def predict_category(resumes_data, selected_category,max_sequence_length):
63
  return ranks
64
 
65
  def load_deeprank_model(max_sequence_length):
66
-
 
67
  model = Sequential()
68
 
69
  model.add(Embedding(input_dim=vocab_size, output_dim=128, input_length=max_sequence_length))
@@ -72,7 +73,6 @@ def load_deeprank_model(max_sequence_length):
72
  model.add(LSTM(64))
73
  model.add(Dense(num_classes, activation='softmax'))
74
  model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy'])
75
- model.load_weights('deeprank_model.h5')
76
  return model
77
 
78
  def main():
 
63
  return ranks
64
 
65
  def load_deeprank_model(max_sequence_length):
66
+
67
+ model.load_weights('deeprank_model.h5')
68
  model = Sequential()
69
 
70
  model.add(Embedding(input_dim=vocab_size, output_dim=128, input_length=max_sequence_length))
 
73
  model.add(LSTM(64))
74
  model.add(Dense(num_classes, activation='softmax'))
75
  model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy'])
 
76
  return model
77
 
78
  def main():