Spaces:
Sleeping
Sleeping
srijaydeshpande
commited on
Update app.py
Browse files
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,
|