Spaces:

jonathang
/

Protein-Family-Ensemble

Runtime error

jonathang commited on Jan 27, 2023

Commit

625231b

•

1 Parent(s): fe37421

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -68,7 +68,7 @@ def get_model():
     model2 = Model(inputs=x_input, outputs=x_output)
     model2.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy'])
-    weights = cached_download(hf_hub_url("jonathang", "Protein_Family_CNN"))
     model2.load_weights(weights)
     return model2

     model2 = Model(inputs=x_input, outputs=x_output)
     model2.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy'])
+    weights = cached_download(hf_hub_url("jonathang/Protein_Family_CNN", 'model2.h5'))
     model2.load_weights(weights)
     return model2