Update main.py

main.py

@@ -20,8 +20,6 @@ ckpt_path = "./pix2seq_reaction_full.ckpt"
 model1 = RxnScribe(ckpt_path, device=torch.device('cpu'))
 device = torch.device('cpu')
 
-API_KEY = os.getenv("API_KEY")
-AZURE_ENDPOINT = os.getenv("AZURE_ENDPOINT")
 
 def ChemEagle(image_path: str) -> dict:
     """
@@ -34,6 +32,9 @@ def ChemEagle(image_path: str) -> dict:
         dict: 整理后的反应数据，包括反应物、产物和反应模板。
     """
     # 初始化 OpenChemIE 模型和 Azure OpenAI 客户端
+    API_KEY = os.getenv("API_KEY")
+    AZURE_ENDPOINT = os.getenv("AZURE_ENDPOINT")
+
     client = AzureOpenAI(
         api_key=API_KEY,
         api_version='2024-06-01',