bstraehle commited on
Commit
702b7af
·
1 Parent(s): a590988

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -22,10 +22,10 @@ vertexai.init(project = project,
22
 
23
  from vertexai.language_models import TextGenerationModel
24
 
25
- generation_model = TextGenerationModel.from_pretrained("text-bison@001")
26
 
27
  def invoke(prompt):
28
- #generation_model = TextGenerationModel.from_pretrained("text-bison@001")
29
  return generation_model.predict(prompt = prompt).text
30
 
31
  gr.close_all()
 
22
 
23
  from vertexai.language_models import TextGenerationModel
24
 
25
+ #generation_model = TextGenerationModel.from_pretrained("text-bison@001")
26
 
27
  def invoke(prompt):
28
+ generation_model = TextGenerationModel.from_pretrained("text-bison@001")
29
  return generation_model.predict(prompt = prompt).text
30
 
31
  gr.close_all()