Update app.py

app.py

@@ -4,18 +4,17 @@ import vertexai, os
 from dotenv import load_dotenv, find_dotenv
 _ = load_dotenv(find_dotenv())
 
-credentials =
-project = 
-openai_api_key = os.environ['OPENAI_API_KEY']
+credentials = os.environ["CREDENTIALS"]
+project = os.environ["PROJECT"]
+
+vertexai.init(project = project, 
+              location = "us-west1", 
+              credentials = credentials)
+
+generation_model = TextGenerationModel.from_pretrained("text-bison@001")
 
 def invoke(prompt):
-    response = openai.ChatCompletion.create(
-        model = "gpt-3.5-turbo",
-        messages = [{"role": "user", "content": prompt}],
-        temperature = 0,
-        max_tokens = 500,
-    )
-    return response.choices[0].message["content"]
+    return generation_model.predict(prompt = prompt).text
 
 gr.close_all()
 demo = gr.Interface(fn=invoke,