Fix wsgi handler name.

PAIR-code · Sep 5, 2024 · 9d39a00 · 9d39a00
1 parent 33f1d77
commit 9d39a00
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lit_nlp/examples/gcp/model_server.py b/lit_nlp/examples/gcp/model_server.py
@@ -56,7 +56,7 @@ def _handler(app, request, environ):
 
   predict_model = models[model_name]
   salience_model = models[f'_{model_name}_salience']
-  tokenize_model = models[f'_{model_name}_tokenizer']
+  tokenize_model = models[f'_{model_name}_tokenize']
 
   handlers = {
       '/predict': predict_model.predict,