Spaces:

billusanda007
/

DeepRank

Sleeping

Jeet Paul commited on Jul 31, 2023

Commit

86733d1

1 Parent(s): 0cdf463

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -72,7 +72,7 @@ def load_deeprank_model(max_sequence_length):
     model.add(LSTM(64))
     model.add(Dense(num_classes, activation='softmax'))
     model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy'])
-    model.load_weights('deeprank_model.h5')  # Replace 'deeprank_model.h5' with your saved model file
     return model
 def main():

     model.add(LSTM(64))
     model.add(Dense(num_classes, activation='softmax'))
     model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy'])
+    model.load_weights('deeprank_model_v2.h5')  # Replace 'deeprank_model.h5' with your saved model file
     return model
 def main():