openfree commited on
Commit
7c4cb16
·
verified ·
1 Parent(s): 79120da

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -0
app.py CHANGED
@@ -16,6 +16,17 @@ import time
16
  from pathlib import Path # 추가
17
  from datetime import datetime, timedelta
18
 
 
 
 
 
 
 
 
 
 
 
 
19
  # 스크린샷 캐시 디렉토리 설정
20
  CACHE_DIR = Path("screenshot_cache")
21
  CACHE_DIR.mkdir(exist_ok=True)
 
16
  from pathlib import Path # 추가
17
  from datetime import datetime, timedelta
18
 
19
+ from huggingface_hub import InferenceClient
20
+ from dotenv import load_dotenv
21
+
22
+ # .env 파일에서 환경 변수 로드
23
+ load_dotenv()
24
+
25
+ # HuggingFace 인퍼런스 클라이언트 설정
26
+ hf_client = InferenceClient(
27
+ "CohereForAI/c4ai-command-r-plus-08-2024",
28
+ token=os.getenv("HF_TOKEN")
29
+ )
30
  # 스크린샷 캐시 디렉토리 설정
31
  CACHE_DIR = Path("screenshot_cache")
32
  CACHE_DIR.mkdir(exist_ok=True)