Update app.py
Browse files
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 |
-
|
17 |
-
|
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
|