Sanket
commited on
Commit
·
ed18734
1
Parent(s):
5225abe
increased input features
Browse files
app.py
CHANGED
@@ -38,7 +38,7 @@ class Generator(nn.Module):
|
|
38 |
|
39 |
# Downsampling
|
40 |
model1 = []
|
41 |
-
in_features =
|
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', '
|
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 |
|