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

Change the default model choice for bb and Mi Sheng.

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