Update app.py

app.py

@@ -22,6 +22,7 @@ from keras.layers import Conv1D, Add, MaxPooling1D, BatchNormalization
 from keras.layers import Embedding, Bidirectional, LSTM, CuDNNLSTM, GlobalMaxPooling1D
 
 import tensorflow as tf
+from huggingface_hub import hf_hub_url, cached_download
 
 
 def residual_block(data, filters, d_rate):
@@ -67,7 +68,8 @@ def get_model():
     
     model2 = Model(inputs=x_input, outputs=x_output)
     model2.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy'])
-    model2.load_weights('model2.h5')
+    weights = cached_download(hf_hub_url("jonathang", "Protein_Family_CNN"))
+    model2.load_weights(weights)
 
     return model2