ZhengPeng7 commited on
Commit
0711aae
·
1 Parent(s): cfb5a5c

Add the missing model_choice in the definition of examples.

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. app_local.py +3 -1
app.py CHANGED
@@ -261,7 +261,7 @@ for idx_example, example in enumerate(examples):
261
  else:
262
  example_resolution = '1024x1024'
263
  model_choice = 'General'
264
- examples[idx_example] = examples[idx_example] + [example_resolution, ]
265
 
266
  examples_url = [
267
  ['https://hips.hearstapps.com/hmg-prod/images/gettyimages-1229892983-square.jpg'],
 
261
  else:
262
  example_resolution = '1024x1024'
263
  model_choice = 'General'
264
+ examples[idx_example] = examples[idx_example] + [example_resolution, model_choice]
265
 
266
  examples_url = [
267
  ['https://hips.hearstapps.com/hmg-prod/images/gettyimages-1229892983-square.jpg'],
app_local.py CHANGED
@@ -253,9 +253,11 @@ examples = [[_] for _ in glob('examples/*')][:]
253
  for idx_example, example in enumerate(examples):
254
  if 'My_' in example[0]:
255
  example_resolution = '2048x2048'
 
256
  else:
257
  example_resolution = '1024x1024'
258
- examples[idx_example] = examples[idx_example] + [example_resolution, 'General']
 
259
 
260
  examples_url = [
261
  ['https://hips.hearstapps.com/hmg-prod/images/gettyimages-1229892983-square.jpg'],
 
253
  for idx_example, example in enumerate(examples):
254
  if 'My_' in example[0]:
255
  example_resolution = '2048x2048'
256
+ model_choice = 'Matting-HR'
257
  else:
258
  example_resolution = '1024x1024'
259
+ model_choice = 'General'
260
+ examples[idx_example] = examples[idx_example] + [example_resolution, model_choice]
261
 
262
  examples_url = [
263
  ['https://hips.hearstapps.com/hmg-prod/images/gettyimages-1229892983-square.jpg'],