billusanda007 commited on
Commit
807012c
·
verified ·
1 Parent(s): d8a8115

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -41,7 +41,7 @@ def predict_category(resumes_data, selected_category,max_sequence_length):
41
  tokenized_text = tokenizer.texts_to_sequences(resumes_text)
42
 
43
 
44
- max_sequence_length = 500 # Assuming maximum sequence length of 500 words
45
  padded_text = pad_sequences(tokenized_text, maxlen=max_sequence_length)
46
 
47
 
@@ -72,7 +72,7 @@ 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_v2.h5') # Replace 'deeprank_model.h5' with your saved model file
76
  return model
77
 
78
  def main():
 
41
  tokenized_text = tokenizer.texts_to_sequences(resumes_text)
42
 
43
 
44
+ max_sequence_length = 500
45
  padded_text = pad_sequences(tokenized_text, maxlen=max_sequence_length)
46
 
47
 
 
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():