Abijith commited on
Commit
fe5d1bf
·
1 Parent(s): 1a5fb0f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -149,7 +149,7 @@ def inference_scribble(image):
149
  return [Image.fromarray((mask_colors * 0.6 + test_image * 0.4).astype('uint8'), 'RGB'),
150
  Image.fromarray((mask_colors ).astype('uint8'), 'RGB')]
151
 
152
- main_scribble = gr.Interface(
153
  fn=inference_scribble,
154
  inputs=
155
  gr.ImageMask(label="[Stroke] Draw on Image", type='pil'),
@@ -165,4 +165,4 @@ main_scribble = gr.Interface(
165
  "./cardamage_example/0206.JPEG"
166
  ]
167
  )
168
- main_scribble.launch()
 
149
  return [Image.fromarray((mask_colors * 0.6 + test_image * 0.4).astype('uint8'), 'RGB'),
150
  Image.fromarray((mask_colors ).astype('uint8'), 'RGB')]
151
 
152
+ demo = gr.Interface(
153
  fn=inference_scribble,
154
  inputs=
155
  gr.ImageMask(label="[Stroke] Draw on Image", type='pil'),
 
165
  "./cardamage_example/0206.JPEG"
166
  ]
167
  )
168
+ demo.launch()