Sqxww commited on
Commit
db5909d
·
1 Parent(s): 58bc2cd

update extract model

Browse files
Files changed (1) hide show
  1. app.py +9 -5
app.py CHANGED
@@ -4,17 +4,21 @@ import requests
4
  prefix = "https://smartfeed-custom-tools.hf.space/gradio_api/file="
5
 
6
  extract_model_tag_list = [
7
- "o3",
 
8
  "o3(poe)",
9
- "gpt-5",
 
10
  "o4-mini(openrouter)",
11
  "o4-mini-high(openrouter)",
12
  ]
13
 
14
  extract_model_map = {
15
- "o3": "o3-2025-04-16",
 
16
  "o3(poe)": "o3(poe)",
17
- "gpt-5": "gpt-5-2025-08-07",
 
18
  "o4-mini(openrouter)": "openai/o4-mini",
19
  "o4-mini-high(openrouter)": "openai/o4-mini-high",
20
  }
@@ -145,7 +149,7 @@ with gr.Blocks() as demo:
145
  prompt_count = gr.Number(label="Prompt Count", value=4, precision=0)
146
  with gr.Column():
147
  image_count = gr.Number(label="Image Count", value=3, precision=0)
148
- extract_model = gr.Dropdown(choices=extract_model_tag_list, label="Extract Model", value="o3")
149
  extract_prompt = gr.Textbox(lines=2, label="Extracted Prompt", value="Accurately extract the hairstyle and makeup from this photo into a prompt especially excluding the backgrounds. And Turn to the side, wearing deep blue Off-shoulder Knitted top. I want to replicate it perfectly. need to be a complete paragraph")
150
  prompt_prefix = gr.Textbox(lines=1, label="Prompt Prefix", value="change the backgrounds to match this:")
151
 
 
4
  prefix = "https://smartfeed-custom-tools.hf.space/gradio_api/file="
5
 
6
  extract_model_tag_list = [
7
+ "o3(chat)",
8
+ "o3(responses)",
9
  "o3(poe)",
10
+ "gpt-5(chat)",
11
+ "gpt-5(responses)",
12
  "o4-mini(openrouter)",
13
  "o4-mini-high(openrouter)",
14
  ]
15
 
16
  extract_model_map = {
17
+ "o3(chat)": "o3-2025-04-16",
18
+ "o3(responses)": "o3-2025-04-16(responses)",
19
  "o3(poe)": "o3(poe)",
20
+ "gpt-5(chat)": "gpt-5-2025-08-07",
21
+ "gpt-5(responses)": "gpt-5-2025-08-07(responses)",
22
  "o4-mini(openrouter)": "openai/o4-mini",
23
  "o4-mini-high(openrouter)": "openai/o4-mini-high",
24
  }
 
149
  prompt_count = gr.Number(label="Prompt Count", value=4, precision=0)
150
  with gr.Column():
151
  image_count = gr.Number(label="Image Count", value=3, precision=0)
152
+ extract_model = gr.Dropdown(choices=extract_model_tag_list, label="Extract Model", value="o3(chat)")
153
  extract_prompt = gr.Textbox(lines=2, label="Extracted Prompt", value="Accurately extract the hairstyle and makeup from this photo into a prompt especially excluding the backgrounds. And Turn to the side, wearing deep blue Off-shoulder Knitted top. I want to replicate it perfectly. need to be a complete paragraph")
154
  prompt_prefix = gr.Textbox(lines=1, label="Prompt Prefix", value="change the backgrounds to match this:")
155