Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -100,11 +100,11 @@ def main():
|
|
100 |
|
101 |
with st.sidebar:
|
102 |
st_lottie(lottie_penguin, height=200)
|
103 |
-
choose = option_menu(
|
104 |
-
icons=['
|
105 |
-
menu_icon="
|
106 |
styles={
|
107 |
-
"container": {"padding": ".0rem", "font-size": "14px"
|
108 |
"nav-link-selected": {"color": "#000000", "font-size": "16px"},
|
109 |
}
|
110 |
)
|
@@ -112,19 +112,15 @@ def main():
|
|
112 |
"""
|
113 |
___
|
114 |
<p style='text-align: center'>
|
115 |
-
FastGAN is
|
|
|
|
|
116 |
</p>
|
117 |
-
<p style='text-align: center'>
|
118 |
-
Model training and
|
119 |
<br/>
|
120 |
<a href="https://huggingface.co/vumichien" target="_blank">Chien Vu</a> | <a href="https://huggingface.co/geninhu" target="_blank">Nhu Hoang</a>
|
121 |
<br/>
|
122 |
-
</p>
|
123 |
-
|
124 |
-
<p style='text-align: center'>
|
125 |
-
based on
|
126 |
-
<br/>
|
127 |
-
<a href="https://github.com/silentz/Towards-Faster-And-Stabilized-GAN-Training-For-High-Fidelity-Few-Shot-Image-Synthesis" target="_blank">FastGAN model</a> | <a href="https://arxiv.org/abs/2101.04775" target="_blank">Article</a>
|
128 |
</p>
|
129 |
""",
|
130 |
unsafe_allow_html=True,
|
@@ -133,7 +129,7 @@ def main():
|
|
133 |
if choose == 'Model Gallery':
|
134 |
st.header("Welcome to FastGAN")
|
135 |
show_model_summary(True)
|
136 |
-
elif choose == '
|
137 |
st.header("Generate images")
|
138 |
col11, col12, col13 = st.columns([3,3.5,3.5])
|
139 |
with col11:
|
@@ -167,8 +163,8 @@ def main():
|
|
167 |
else:
|
168 |
st.image(gan_images[1:], width=150)
|
169 |
|
170 |
-
elif choose == '
|
171 |
-
st.header("
|
172 |
st.markdown(
|
173 |
"""
|
174 |
<p style='text-align: left'>
|
|
|
100 |
|
101 |
with st.sidebar:
|
102 |
st_lottie(lottie_penguin, height=200)
|
103 |
+
choose = option_menu("FastGAN", ["Model Gallery", "Generate images", "Mix style"],
|
104 |
+
icons=['collection', 'file-plus', 'intersect'],
|
105 |
+
menu_icon="infinity", default_index=0,
|
106 |
styles={
|
107 |
+
"container": {"padding": ".0rem", "font-size": "14px"},
|
108 |
"nav-link-selected": {"color": "#000000", "font-size": "16px"},
|
109 |
}
|
110 |
)
|
|
|
112 |
"""
|
113 |
___
|
114 |
<p style='text-align: center'>
|
115 |
+
FastGAN is a few-shot GAN model trained on high-fidelity images which requires less computation resource and samples for training.
|
116 |
+
<br/>
|
117 |
+
<a href="https://arxiv.org/abs/2101.04775" target="_blank">Article</a>
|
118 |
</p>
|
119 |
+
<p style='text-align: center; font-size: 14px;'>
|
120 |
+
Model training and Spaces creating by
|
121 |
<br/>
|
122 |
<a href="https://huggingface.co/vumichien" target="_blank">Chien Vu</a> | <a href="https://huggingface.co/geninhu" target="_blank">Nhu Hoang</a>
|
123 |
<br/>
|
|
|
|
|
|
|
|
|
|
|
|
|
124 |
</p>
|
125 |
""",
|
126 |
unsafe_allow_html=True,
|
|
|
129 |
if choose == 'Model Gallery':
|
130 |
st.header("Welcome to FastGAN")
|
131 |
show_model_summary(True)
|
132 |
+
elif choose == 'Generate images':
|
133 |
st.header("Generate images")
|
134 |
col11, col12, col13 = st.columns([3,3.5,3.5])
|
135 |
with col11:
|
|
|
163 |
else:
|
164 |
st.image(gan_images[1:], width=150)
|
165 |
|
166 |
+
elif choose == 'Mix style':
|
167 |
+
st.header("Mix style")
|
168 |
st.markdown(
|
169 |
"""
|
170 |
<p style='text-align: left'>
|