Update app.py

app.py

@@ -4,7 +4,7 @@ import json, os, vertexai, wandb
 from dotenv import load_dotenv, find_dotenv
 _ = load_dotenv(find_dotenv())
 
-//credentials = os.environ["CREDENTIALS"]
+#credentials = os.environ["CREDENTIALS"]
 project = os.environ["PROJECT"]
 wandb_api_key = os.environ["WANDB_API_KEY"]
 
@@ -16,23 +16,24 @@ wandb.login(key = wandb_api_key)
 wandb.init(project = "vertext-ai-txt", config = config)
 config = wandb.config
 
-//credentials = json.loads(credentials)
+#credentials = json.loads(credentials)
 
 from google.oauth2 import service_account
-//credentials = service_account.Credentials.from_service_account_info(credentials)
+#credentials = service_account.Credentials.from_service_account_info(credentials)
 
-//if credentials.expired:
-//    credentials.refresh(Request())
+#if credentials.expired:
+#    credentials.refresh(Request())
 
-//vertexai.init(project = project, 
-//              location = "us-central1",
-//              credentials = credentials)
+#vertexai.init(project = project, 
+#              location = "us-central1",
+#              credentials = credentials)
 
 from vertexai.language_models import TextGenerationModel
 generation_model = TextGenerationModel.from_pretrained("text-bison@001")
 
 def invoke(prompt, credentials):
     //
+    credentials = json.loads(credentials)
     credentials = service_account.Credentials.from_service_account_info(credentials)
     if credentials.expired:
         credentials.refresh(Request())