Update app.py
Browse files
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", "
|
| 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="
|
| 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-
|
| 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", "
|
| 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を返します
|