tomonori1120 commited on
Commit
70f0e88
·
verified ·
1 Parent(s): f19b307

feat: hf_tokenをオプション化 - 環境変数HF_TOKENから自動取得

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -417,7 +417,7 @@ def extend_existing_video(input_image_gallery: List[str], prompt_input: str, out
417
  # =====================================================
418
  def mcp_generate_image(
419
  prompt: str,
420
- hf_token: str,
421
  aspect_ratio: str = "Auto",
422
  model: str = "Nano Banana",
423
  resolution: str = "1K"
@@ -435,6 +435,10 @@ def mcp_generate_image(
435
  Returns:
436
  Path to the generated image file
437
  """
 
 
 
 
438
  if not verify_pro_status(hf_token):
439
  raise gr.Error("Access Denied. A valid Hugging Face PRO token is required.")
440
 
 
417
  # =====================================================
418
  def mcp_generate_image(
419
  prompt: str,
420
+ hf_token: str = "",
421
  aspect_ratio: str = "Auto",
422
  model: str = "Nano Banana",
423
  resolution: str = "1K"
 
435
  Returns:
436
  Path to the generated image file
437
  """
438
+ # トークン未指定の場合は環境変数から自動取得
439
+ if not hf_token:
440
+ hf_token = os.getenv("HF_TOKEN", "")
441
+
442
  if not verify_pro_status(hf_token):
443
  raise gr.Error("Access Denied. A valid Hugging Face PRO token is required.")
444