Update app.py
Browse files
app.py
CHANGED
|
@@ -25,7 +25,7 @@ def _clone_and_checkout(repo_url: str, local_path: str, version: str) -> Reposit
|
|
| 25 |
return repository
|
| 26 |
|
| 27 |
_ = _clone_and_checkout(model_repo_url, local_path, model_version)
|
| 28 |
-
|
| 29 |
# gpt_lm_predict_fn = model.signatures["serving_default"]
|
| 30 |
|
| 31 |
with gr.Blocks() as demo:
|
|
|
|
| 25 |
return repository
|
| 26 |
|
| 27 |
_ = _clone_and_checkout(model_repo_url, local_path, model_version)
|
| 28 |
+
model = tf.saved_model.load(local_path, tags=[tag_constants.SERVING])
|
| 29 |
# gpt_lm_predict_fn = model.signatures["serving_default"]
|
| 30 |
|
| 31 |
with gr.Blocks() as demo:
|