endrol commited on
Commit
96f2754
·
1 Parent(s): c7b35f4

add dceplus model

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -10,12 +10,14 @@ from PIL import Image
10
  SR_API = os.getenv("sr_api")
11
  ENHANCE_API = os.getenv("enhance_api")
12
  ZERODCE_API = os.getenv("zerodce_api")
 
13
 
14
  # Get API endpoints from environment
15
  API_ENDPOINTS = {
16
  "SR API": SR_API,
17
  "ENHANCE API": ENHANCE_API,
18
- "ZERODCE API": ZERODCE_API
 
19
  }
20
 
21
  def apply_super_resolution(input_image, scale_factor, tile_size, api_choice):
@@ -94,7 +96,7 @@ def main():
94
  gr.Markdown("### Enhancement Settings")
95
 
96
  api_dropdown = gr.Dropdown(
97
- choices=["SR API", "ENHANCE API", "ZERODCE API"],
98
  value="SR API",
99
  label="API Choice",
100
  info="Choose which enhancement API to use"
 
10
  SR_API = os.getenv("sr_api")
11
  ENHANCE_API = os.getenv("enhance_api")
12
  ZERODCE_API = os.getenv("zerodce_api")
13
+ ZERODCE_PLUS_API = os.getenv("zerodce_plus_api")
14
 
15
  # Get API endpoints from environment
16
  API_ENDPOINTS = {
17
  "SR API": SR_API,
18
  "ENHANCE API": ENHANCE_API,
19
+ "ZERODCE API": ZERODCE_API,
20
+ "ZERODCE++ API": ZERODCE_PLUS_API
21
  }
22
 
23
  def apply_super_resolution(input_image, scale_factor, tile_size, api_choice):
 
96
  gr.Markdown("### Enhancement Settings")
97
 
98
  api_dropdown = gr.Dropdown(
99
+ choices=["SR API", "ENHANCE API", "ZERODCE API", "ZERODCE++ API"],
100
  value="SR API",
101
  label="API Choice",
102
  info="Choose which enhancement API to use"