davidlee831117 commited on
Commit
f89aa16
·
verified ·
1 Parent(s): 0813225

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -9
app.py CHANGED
@@ -13,7 +13,6 @@ from google.oauth2 import service_account
13
 
14
  # 由於您的環境可能沒有服務帳號金鑰,我們將直接使用 GEMINI_API_KEY
15
  # 如果在 Hugging Face Space 中,這通常會從環境變數中自動加載
16
- # 您仍然需要設置 GEMINI_API_KEY 作為您的 API 金鑰
17
 
18
  # 初始化 AI Platform 服務客戶端
19
  def initialize_ai_platform(api_key):
@@ -21,11 +20,7 @@ def initialize_ai_platform(api_key):
21
  PROJECT_ID = "YOUR_GCP_PROJECT_ID" # 這裡需要您提供真實的GCP專案ID
22
  LOCATION = "us-central1"
23
 
24
- # 這裡的認證方式取決於您的環境。在 Hugging Face Space 中,我們無法直接使用服務帳號金鑰。
25
- # 這裡我們用一個假定的方式來適應 Gradio 環境,但實際上您可能需要在 GCP 內部運行。
26
- # 如果您只是使用 API 金鑰,這部分會更簡單,但 'google-cloud-aiplatform' 函式庫
27
- # 主要設計用於 GCP 環境,所以這可能會比較複雜。
28
-
29
  # 創建一個不帶服務帳號憑證的客戶端,這會讓它嘗試從環境中自動尋找憑證
30
  try:
31
  aiplatform.init(project=PROJECT_ID, location=LOCATION)
@@ -34,7 +29,7 @@ def initialize_ai_platform(api_key):
34
  print(f"Error initializing AI Platform: {e}")
35
  return False
36
 
37
- # 全局變數來儲存 API 金鑰,以確保與之前的版本保持一致
38
  GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
39
 
40
  if not GEMINI_API_KEY:
@@ -49,8 +44,8 @@ if not GEMINI_API_KEY:
49
  raise ValueError("ERROR: GEMINI_API_KEY environment variable is not set. Please set it correctly.")
50
 
51
  # 這裡我們將創建一個通用客戶端來處理 API 呼叫
52
- # 由於 'google-cloud-aiplatform' 更適合 GCP,我們將使用 `requests` 來模擬
53
- # 直接呼叫 API,以適應您在 Hugging Face Space 的需求。
54
  class NanoBananaImageGenerator:
55
  def __init__(self, api_key):
56
  self.api_key = api_key
 
13
 
14
  # 由於您的環境可能沒有服務帳號金鑰,我們將直接使用 GEMINI_API_KEY
15
  # 如果在 Hugging Face Space 中,這通常會從環境變數中自動加載
 
16
 
17
  # 初始化 AI Platform 服務客戶端
18
  def initialize_ai_platform(api_key):
 
20
  PROJECT_ID = "YOUR_GCP_PROJECT_ID" # 這裡需要您提供真實的GCP專案ID
21
  LOCATION = "us-central1"
22
 
23
+ # 這裡的認證方式取決於您的環境。
 
 
 
 
24
  # 創建一個不帶服務帳號憑證的客戶端,這會讓它嘗試從環境中自動尋找憑證
25
  try:
26
  aiplatform.init(project=PROJECT_ID, location=LOCATION)
 
29
  print(f"Error initializing AI Platform: {e}")
30
  return False
31
 
32
+ # 全局變數來儲存 API 金鑰
33
  GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
34
 
35
  if not GEMINI_API_KEY:
 
44
  raise ValueError("ERROR: GEMINI_API_KEY environment variable is not set. Please set it correctly.")
45
 
46
  # 這裡我們將創建一個通用客戶端來處理 API 呼叫
47
+ # 由於 'google-cloud-aiplatform' 函式庫主要設計用於 GCP 環境,我們將使用 `requests`
48
+ # 來模擬直接呼叫 API,以適應您在 Hugging Face Space 的需求。
49
  class NanoBananaImageGenerator:
50
  def __init__(self, api_key):
51
  self.api_key = api_key