bstraehle commited on
Commit
7de5168
·
1 Parent(s): d69bc6d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -7,14 +7,15 @@ _ = load_dotenv(find_dotenv())
7
  credentials = os.environ["CREDENTIALS"]
8
  project = os.environ["PROJECT"]
9
 
 
 
10
  from google.oauth2 import service_account
11
 
12
- credentials = json.loads(credentials)
13
  credentials = service_account.Credentials.from_service_account_info(credentials)
14
  print("###1")
15
 
16
- #if credentials.expired:
17
- # credentials.refresh(Request())
18
 
19
  vertexai.init(project = project,
20
  location = "us-west1",
@@ -24,7 +25,7 @@ print("###2")
24
  #from vertexai.language_models import TextGenerationModel
25
 
26
  #generation_model = TextGenerationModel.from_pretrained("text-bison@001")
27
- print("###3")
28
 
29
  def invoke(prompt):
30
  return generation_model.predict(prompt = prompt).text
 
7
  credentials = os.environ["CREDENTIALS"]
8
  project = os.environ["PROJECT"]
9
 
10
+ credentials = json.loads(credentials)
11
+
12
  from google.oauth2 import service_account
13
 
 
14
  credentials = service_account.Credentials.from_service_account_info(credentials)
15
  print("###1")
16
 
17
+ if credentials.expired:
18
+ credentials.refresh(Request())
19
 
20
  vertexai.init(project = project,
21
  location = "us-west1",
 
25
  #from vertexai.language_models import TextGenerationModel
26
 
27
  #generation_model = TextGenerationModel.from_pretrained("text-bison@001")
28
+ #print("###3")
29
 
30
  def invoke(prompt):
31
  return generation_model.predict(prompt = prompt).text