Update app.py

app.py

@@ -4,20 +4,14 @@ import os, vertexai
 from dotenv import load_dotenv, find_dotenv
 _ = load_dotenv(find_dotenv())
 
-from vertexai.language_models import TextGenerationModel
-
-print(1)
-
 credentials = os.environ["CREDENTIALS"]
 project = os.environ["PROJECT"]
 
-print(2)
-
 vertexai.init(project = project, 
               location = "us-west1", 
               credentials = credentials)
 
-print(3)
+from vertexai.language_models import TextGenerationModel
 
 generation_model = TextGenerationModel.from_pretrained("text-bison@001")