Jeet Paul commited on
Commit
bbbf59d
·
1 Parent(s): c1ae8aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -62,7 +62,7 @@ def predict_category(resumes_data, selected_category):
62
 
63
  return ranks
64
 
65
- def load_deeprank_model():
66
  # Load the saved DeepRank model
67
  model = Sequential()
68
  # Add layers to the model (example architecture, adjust as needed)
@@ -111,4 +111,6 @@ if __name__ == '__main__':
111
  vocab_size = len(tokenizer.word_index) + 1
112
  num_classes = len(label.classes_)
113
 
 
 
114
  main()
 
62
 
63
  return ranks
64
 
65
+ def load_deeprank_model(max_sequence_length):
66
  # Load the saved DeepRank model
67
  model = Sequential()
68
  # Add layers to the model (example architecture, adjust as needed)
 
111
  vocab_size = len(tokenizer.word_index) + 1
112
  num_classes = len(label.classes_)
113
 
114
+ max_sequence_length = 500
115
+
116
  main()