aidlab/sora_python
7
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

TASK:lookbook上传

Browse Source
This commit is contained in:
shahaibo
2024-10-22 00:01:27 +08:00
parent 26360dd2b5
commit 5a26642280
1 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
app/service/lookbooks/config/config.py
View File

@@ -1,15 +1,15 @@
# config.py
import os
import platform
if platform.system() == 'Linux':
__import__('pysqlite3')
import sys
sys.modules['sqlite3'] = sys.modules.pop('pysqlite3')
# import platform
# if platform.system() == 'Linux':
# __import__('pysqlite3')
# import sys
# sys.modules['sqlite3'] = sys.modules.pop('pysqlite3')
import chromadb
from langchain_openai import OpenAIEmbeddings
from langchain_chroma import Chroma
import tritonclient.grpc as grpcclient
from minio import Minio
# import tritonclient.grpc as grpcclient
# from minio import Minio
# OpenAI settings
OPENAI_API_KEY = "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"
@@ -26,7 +26,7 @@ MINIO_SECURE = False
MINIO_ACCESS = "e8zc55mzDOh4IzRrZ9Oa"
MINIO_SECRET = "uHfqJ7UkwA1PTDGfnA44Hp9ux5YkZTkzZLjeOYhE"
MINIO_BUCKET = "test"
MINIO_CLIENT = Minio(MINIO_URL, access_key=MINIO_ACCESS, secret_key=MINIO_SECRET, secure=MINIO_SECURE)
# MINIO_CLIENT = Minio(MINIO_URL, access_key=MINIO_ACCESS, secret_key=MINIO_SECRET, secure=MINIO_SECURE)
# Set environment variables
os.environ["OPENAI_API_KEY"] = OPENAI_API_KEY
@@ -71,4 +71,4 @@ elif OP_SERVER == "aws":
# os.environ["OOTD_URL"] = "http://18.167.251.121:10001/ootd/ootd_dc"
os.environ["OOTD_URL"] = "https://muskox-many-bluegill.ngrok-free.app/ootd_dc"
triton_client = grpcclient.InferenceServerClient(url=os.environ['GRPCCLIENT_URL'])
# triton_client = grpcclient.InferenceServerClient(url=os.environ['GRPCCLIENT_URL'])