leave-everything commited on
Commit
0eb31e9
·
verified ·
1 Parent(s): 736bda3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -34,7 +34,7 @@ logger = logging.getLogger(__name__)
34
  class HuggingFaceUploader:
35
  """HuggingFace Hubへ画像をアップロードする機能を提供するクラス"""
36
  def __init__(self):
37
- self.repo_id = os.environ.get("HF_REPO_ID", "tomo2chin2/SUPER_TENSAI_JIN")
38
  self.token = os.environ.get("HF_TOKEN", None)
39
  if self.token:
40
  try:
@@ -334,7 +334,7 @@ def load_system_instruction(style="standard"):
334
  logger.info("デフォルトのprompt.txtを読み込みます")
335
 
336
  file_path = hf_hub_download(
337
- repo_id="tomo2chin2/GURAREKOstlyle",
338
  filename="prompt.txt",
339
  repo_type="dataset"
340
  )
@@ -360,7 +360,7 @@ def generate_html_from_text(text, temperature=0.3, style="standard"):
360
  raise ValueError("GEMINI_API_KEY 環境変数が設定されていません")
361
 
362
  # モデル名の取得(環境変数から、なければデフォルト値)
363
- model_name = os.environ.get("GEMINI_MODEL", "gemini-1.5-pro")
364
  logger.info(f"使用するGeminiモデル: {model_name}")
365
 
366
  # Gemini APIの設定
@@ -1114,7 +1114,7 @@ with gr.Blocks(title="Full Page Screenshot (テキスト変換対応)", theme=gr
1114
 
1115
  # 環境変数情報を表示
1116
  gemini_model = os.environ.get("GEMINI_MODEL", "gemini-1.5-pro")
1117
- hf_repo = os.environ.get("HF_REPO_ID", "tomo2chin2/SUPER_TENSAI_JIN")
1118
  gr.Markdown(f"""
1119
  ## APIエンドポイント
1120
  - `/api/screenshot` - HTMLコードからスクリーンショットを生成し、URLを返します
 
34
  class HuggingFaceUploader:
35
  """HuggingFace Hubへ画像をアップロードする機能を提供するクラス"""
36
  def __init__(self):
37
+ self.repo_id = os.environ.get("HF_REPO_ID", "leave-everything/ChoTensaiJinGrareko")
38
  self.token = os.environ.get("HF_TOKEN", None)
39
  if self.token:
40
  try:
 
334
  logger.info("デフォルトのprompt.txtを読み込みます")
335
 
336
  file_path = hf_hub_download(
337
+ repo_id="leave-everything/GURAREKOstyle",
338
  filename="prompt.txt",
339
  repo_type="dataset"
340
  )
 
360
  raise ValueError("GEMINI_API_KEY 環境変数が設定されていません")
361
 
362
  # モデル名の取得(環境変数から、なければデフォルト値)
363
+ model_name = os.environ.get("GEMINI_MODEL", "gemini-flash-latest")
364
  logger.info(f"使用するGeminiモデル: {model_name}")
365
 
366
  # Gemini APIの設定
 
1114
 
1115
  # 環境変数情報を表示
1116
  gemini_model = os.environ.get("GEMINI_MODEL", "gemini-1.5-pro")
1117
+ hf_repo = os.environ.get("HF_REPO_ID", "leave-everything/ChoTensaiJinGrareko")
1118
  gr.Markdown(f"""
1119
  ## APIエンドポイント
1120
  - `/api/screenshot` - HTMLコードからスクリーンショットを生成し、URLを返します