Spaces:
Sleeping
Sleeping
Update agent_langchain.py
Browse files- agent_langchain.py +2 -0
agent_langchain.py
CHANGED
|
@@ -7,6 +7,7 @@ os.environ["TORCH_HOME"] = "/tmp/torch"
|
|
| 7 |
|
| 8 |
import requests
|
| 9 |
import torch
|
|
|
|
| 10 |
import time
|
| 11 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
| 12 |
import numpy as np
|
|
@@ -28,6 +29,7 @@ GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
|
|
| 28 |
ROUTING_URL = os.environ.get("ROUTING_URL")
|
| 29 |
SPACE_URL = os.environ.get("SPACE_URL", "http://localhost:7860")
|
| 30 |
FIREBASE_CREDS_PATH = os.environ.get("FIREBASE_CREDS_PATH")
|
|
|
|
| 31 |
|
| 32 |
# Initialize Firebase
|
| 33 |
db = None
|
|
|
|
| 7 |
|
| 8 |
import requests
|
| 9 |
import torch
|
| 10 |
+
import json
|
| 11 |
import time
|
| 12 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
| 13 |
import numpy as np
|
|
|
|
| 29 |
ROUTING_URL = os.environ.get("ROUTING_URL")
|
| 30 |
SPACE_URL = os.environ.get("SPACE_URL", "http://localhost:7860")
|
| 31 |
FIREBASE_CREDS_PATH = os.environ.get("FIREBASE_CREDS_PATH")
|
| 32 |
+
firebase_creds_json = os.getenv("FIREBASE_CREDS_JSON")
|
| 33 |
|
| 34 |
# Initialize Firebase
|
| 35 |
db = None
|