Spaces:

bstraehle
/

vertex-ai

Sleeping

bstraehle commited on Oct 15, 2023

Commit

545971b

1 Parent(s): b50ab7b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -25,6 +25,7 @@ from vertexai.language_models import TextGenerationModel
 #generation_model = TextGenerationModel.from_pretrained("text-bison@001")
 def invoke(prompt):
     return generation_model.predict(prompt = prompt).text
 gr.close_all()

 #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()