sunilsarolkar commited on
Commit
6dd4ed2
·
verified ·
1 Parent(s): 6363a15

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -150,7 +150,12 @@ def get_translator_model():
150
  translation_model.add(keras.layers.Activation('elu'))
151
  translation_model.add(Dropout(0.2))
152
  translation_model.add(Dense(len(list(expression_mapping.keys())), activation='softmax'))
153
- translation_model.load_weights('isl_model_final.keras')
 
 
 
 
 
154
  return translation_model
155
 
156
 
 
150
  translation_model.add(keras.layers.Activation('elu'))
151
  translation_model.add(Dropout(0.2))
152
  translation_model.add(Dense(len(list(expression_mapping.keys())), activation='softmax'))
153
+ model_file = hf_hub_download(
154
+ repo_id="sunilsarolkar/isl-translation-model",
155
+ filename="isl_model_final.keras"
156
+ )
157
+ translation_model.load_weights(model_file)
158
+ translation_model.load_weights(model_file)
159
  return translation_model
160
 
161