Update app.py

app.py

@@ -16,29 +16,22 @@ 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())
-    vertexai.init(project = project, 
-                  location = "us-central1",
-                  credentials = credentials)
+def invoke(prompt):
     completion = generation_model.predict(prompt = prompt).text
     wandb.log({"prompt": prompt, "completion": completion})
     return completion