Update app.py
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
|