fixed error in PredictionServiceClient calling

PiperOrigin-RevId: 687477199 Change-Id: Ia108098090df74fedc68048004ccabad38548cab
google-deepmind · Oct 19, 2024 · e24a50d · e24a50d
1 parent c7e8777
commit e24a50d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/concordia/language_model/google_cloud_custom_model.py b/concordia/language_model/google_cloud_custom_model.py
@@ -134,8 +134,7 @@ def sample_text(
 
       try:
         response = (
-            client()
-            .predict(
+            client.predict(
                 endpoint=endpoint_name,
                 instances=[{"inputs": prompt}],
                 parameters={