Upload demo.py with huggingface_hub
Browse files
demo.py
CHANGED
|
@@ -5,14 +5,19 @@ from options.option_llm import get_args_parser
|
|
| 5 |
from utils.motion_utils import recover_from_ric, plot_3d_motion
|
| 6 |
from utils.paramUtil import t2m_kinematic_chain
|
| 7 |
import torch
|
|
|
|
|
|
|
|
|
|
| 8 |
|
| 9 |
def motion_agent_demo():
|
| 10 |
# Initialize the client
|
| 11 |
-
client =
|
| 12 |
-
|
| 13 |
-
|
| 14 |
-
|
| 15 |
-
|
|
|
|
|
|
|
| 16 |
|
| 17 |
args = get_args_parser()
|
| 18 |
args.save_dir = "./demo"
|
|
|
|
| 5 |
from utils.motion_utils import recover_from_ric, plot_3d_motion
|
| 6 |
from utils.paramUtil import t2m_kinematic_chain
|
| 7 |
import torch
|
| 8 |
+
from openai import OpenAI
|
| 9 |
+
|
| 10 |
+
|
| 11 |
|
| 12 |
def motion_agent_demo():
|
| 13 |
# Initialize the client
|
| 14 |
+
client = OpenAI(
|
| 15 |
+
api_key=os.environ.get("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"), # This is the default and can be omitted
|
| 16 |
+
)
|
| 17 |
+
|
| 18 |
+
# client = AzureOpenAI(
|
| 19 |
+
# api_key="", # your api key
|
| 20 |
+
# )
|
| 21 |
|
| 22 |
args = get_args_parser()
|
| 23 |
args.save_dir = "./demo"
|