Sanket commited on
Commit
ed18734
·
1 Parent(s): 5225abe

increased input features

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -38,7 +38,7 @@ class Generator(nn.Module):
38
 
39
  # Downsampling
40
  model1 = []
41
- in_features = 64
42
  out_features = in_features*2
43
  for _ in range(2):
44
  model1 += [ nn.Conv2d(in_features, out_features, 3, stride=2, padding=1),
@@ -108,7 +108,7 @@ def predict(input_img, ver):
108
 
109
  title="Image to Line Drawings - Complex and Simple Portraits and Landscapes"
110
  examples=[
111
- ['01.jpg', 'Simple Lines'], ['02.jpg', 'Simple Lines'], ['03.jpg', 'Simple Lines'],
112
  ['04.jpg', 'Simple Lines'], ['05.jpg', 'Simple Lines'],
113
  ]
114
 
 
38
 
39
  # Downsampling
40
  model1 = []
41
+ in_features = 256
42
  out_features = in_features*2
43
  for _ in range(2):
44
  model1 += [ nn.Conv2d(in_features, out_features, 3, stride=2, padding=1),
 
108
 
109
  title="Image to Line Drawings - Complex and Simple Portraits and Landscapes"
110
  examples=[
111
+ ['01.jpg', 'Complex Lines'], ['02.jpg', 'Complex Lines'], ['03.jpg', 'Simple Lines'],
112
  ['04.jpg', 'Simple Lines'], ['05.jpg', 'Simple Lines'],
113
  ]
114