srijaydeshpande commited on
Commit
1c3571e
·
verified ·
1 Parent(s): 7d7c2bc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -277,6 +277,10 @@ model = SPADEResNet(input_nc=3, output_nc=6)
277
  model = nn.DataParallel(model)
278
  model = model.cuda()
279
  model.load_state_dict(torch.load(model_path), strict=True)
 
 
 
 
280
 
281
  demo = gr.Interface(
282
  segment_image,
 
277
  model = nn.DataParallel(model)
278
  model = model.cuda()
279
  model.load_state_dict(torch.load(model_path), strict=True)
280
+ examples = [
281
+ ["sample1.png"],
282
+ ["sample2.png"]
283
+ ]
284
 
285
  demo = gr.Interface(
286
  segment_image,