Update app.py

app.py

@@ -12,7 +12,6 @@ credentials = json.loads(credentials)
 from google.oauth2 import service_account
 
 credentials = service_account.Credentials.from_service_account_info(credentials)
-print("###1")
 
 if credentials.expired:
     credentials.refresh(Request())
@@ -20,12 +19,10 @@ if credentials.expired:
 vertexai.init(project = project, 
               location = "us-west1",
               credentials = credentials)
-print("###2")
 
-#from vertexai.language_models import TextGenerationModel
+from vertexai.language_models import TextGenerationModel
 
 #generation_model = TextGenerationModel.from_pretrained("text-bison@001")
-#print("###3")
 
 def invoke(prompt):
     return generation_model.predict(prompt = prompt).text