Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -181,6 +181,11 @@ def generate_html_from_text(text: str, temperature=0.5, style="standard") -> str
|
|
| 181 |
candidate_count=1
|
| 182 |
)
|
| 183 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 184 |
# Generate content
|
| 185 |
response = client.models.generate_content(
|
| 186 |
model=model_name,
|
|
@@ -341,8 +346,14 @@ with gr.Blocks(title="Full Page Screenshot (テキスト変換対応)", theme=gr
|
|
| 341 |
|
| 342 |
btn.click(process_input, [mode, text, ext, temp, trim, style], out)
|
| 343 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 344 |
gr.Markdown(f"**API** `/api/screenshot`, `/api/text-to-screenshot` "
|
| 345 |
-
f"使用モデル: `{
|
| 346 |
|
| 347 |
# ===============================================================
|
| 348 |
# Gradio を /gradio にマウントし、明示リダイレクトを追加
|
|
|
|
| 181 |
candidate_count=1
|
| 182 |
)
|
| 183 |
|
| 184 |
+
# 新機能: Gemini 2.5 Flash Preview モデルの場合、思考モードをオフにする
|
| 185 |
+
if model_name == "gemini-2.5-flash-preview-04-17":
|
| 186 |
+
logger.info("gemini-2.5-flash-preview-04-17 モデル検出: 思考モードをオフに設定")
|
| 187 |
+
config.thinking_config = types.ThinkingConfig(thinking_budget=0)
|
| 188 |
+
|
| 189 |
# Generate content
|
| 190 |
response = client.models.generate_content(
|
| 191 |
model=model_name,
|
|
|
|
| 346 |
|
| 347 |
btn.click(process_input, [mode, text, ext, temp, trim, style], out)
|
| 348 |
|
| 349 |
+
# モデル名を表示し、Gemini 2.5の場合は思考モードのステータスも表示
|
| 350 |
+
model_name = os.getenv('GEMINI_MODEL', 'gemini-1.5-pro')
|
| 351 |
+
thinking_status = ""
|
| 352 |
+
if model_name == "gemini-2.5-flash-preview-04-17":
|
| 353 |
+
thinking_status = "(思考モード: オフ)"
|
| 354 |
+
|
| 355 |
gr.Markdown(f"**API** `/api/screenshot`, `/api/text-to-screenshot` "
|
| 356 |
+
f"使用モデル: `{model_name}` {thinking_status}")
|
| 357 |
|
| 358 |
# ===============================================================
|
| 359 |
# Gradio を /gradio にマウントし、明示リダイレクトを追加
|