jonathang commited on
Commit
0bc7b8a
1 Parent(s): 994d177

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -91,14 +91,14 @@ def get_model():
91
 
92
  model2 = Model(inputs=x_input, outputs=x_output)
93
  model2.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy'])
94
- weights = cached_download(hf_hub_url("jonathang/Protein_Family_CNN", 'model2.h5'))
95
  model2.load_weights(weights)
96
 
97
  return model2
98
 
99
 
100
  model = get_model()
101
- mappings_path = cached_download(hf_hub_url("jonathang/Protein_Family_CNN", 'prot_mappings.json'))
102
  with open(mappings_path) as f:
103
  prot_mappings = json.load(f)
104
 
 
91
 
92
  model2 = Model(inputs=x_input, outputs=x_output)
93
  model2.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy'])
94
+ weights = cached_download(hf_hub_url("jonathang/Protein_Family_Models", 'model2.h5'))
95
  model2.load_weights(weights)
96
 
97
  return model2
98
 
99
 
100
  model = get_model()
101
+ mappings_path = cached_download(hf_hub_url("jonathang/Protein_Family_Models", 'prot_mappings.json'))
102
  with open(mappings_path) as f:
103
  prot_mappings = json.load(f)
104