Update app.py

app.py

@@ -10,7 +10,6 @@ project = os.environ["PROJECT"]
 credentials = json.loads(credentials)
 
 from google.oauth2 import service_account
-
 credentials = service_account.Credentials.from_service_account_info(credentials)
 
 if credentials.expired:
@@ -21,11 +20,9 @@ vertexai.init(project = project,
               credentials = credentials)
 
 from vertexai.language_models import TextGenerationModel
-
-#generation_model = TextGenerationModel.from_pretrained("text-bison@001")
+generation_model = TextGenerationModel.from_pretrained("text-bison@001")
 
 def invoke(prompt):
-    generation_model = TextGenerationModel.from_pretrained("text-bison@001")
     return generation_model.predict(prompt = prompt).text
 
 gr.close_all()