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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -64,15 +64,15 @@ def predict_category(resumes_data, selected_category,max_sequence_length):
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))
70
- #model.add(Conv1D(filters=128, kernel_size=5, activation='relu'))
71
- #model.add(MaxPooling1D(pool_size=2))
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_v2.h5')
76
  return model
77
 
78
  def main():
 
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))
70
+ model.add(Conv1D(filters=128, kernel_size=5, activation='relu'))
71
+ model.add(MaxPooling1D(pool_size=2))
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():