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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -64,14 +64,14 @@ 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
 
 
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