vertex ai

app.py

@@ -10,6 +10,23 @@ desired_language = "en"
 response_generator_pipe = TextGenerationPipeline(max_length=max_answer_length)
 
 
+import vertexai
+from vertexai.preview.language_models import ChatModel, InputOutputTextPair
+
+vertexai.init(project="youtube-live-checker-245701", location="us-central1")
+chat_model = ChatModel.from_pretrained("chat-bison@001")
+parameters = {
+    "temperature": 0.2,
+    "max_output_tokens": 256,
+    "top_p": 0.8,
+    "top_k": 40
+}
+chat = chat_model.start_chat(
+    context="""Act as Santa and respond like I am a 10 year old child.""",
+)
+
+
+
 def main(audio: object):
     """Calls functions for deploying gradio app.
 

requirements.txt

@@ -4,4 +4,6 @@ gradio==3.14.0
 SpeechRecognition==3.9.0
 mtranslate==1.8
 gTTS==2.3.0
-transformers==4.25.1
+transformers==4.25.1
+google-cloud-aiplatform
+vertexai==1.28.1