geninhu commited on
Commit
8d454d8
1 Parent(s): 7734792

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +39 -23
app.py CHANGED
@@ -18,9 +18,11 @@ device = 'cuda' if torch.cuda.is_available() else 'cpu'
18
  model_name = {
19
  "aurora": 'huggan/fastgan-few-shot-aurora',
20
  "painting": 'huggan/fastgan-few-shot-painting',
21
- #"painting":"geninhu/fastgan-few-shot-art",
22
  "shell": 'huggan/fastgan-few-shot-shells',
23
  "fauvism": 'huggan/fastgan-few-shot-fauvism-still-life',
 
 
 
24
  }
25
 
26
  #@st.cache(allow_output_mutation=True)
@@ -54,6 +56,34 @@ def load_lottieurl(url: str):
54
  return None
55
  return r.json()
56
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
57
 
58
  def main():
59
 
@@ -91,36 +121,22 @@ def main():
91
  )
92
 
93
  st.header("Welcome to FastGAN")
 
94
 
95
- col1, col2, col3, col4 = st.columns([3,3,3,3])
96
- with col1:
97
- st.markdown('Fauvism GAN [model](https://huggingface.co/huggan/fastgan-few-shot-fauvism-still-life)', unsafe_allow_html=True)
98
- st.image('fauvism.png', width=220)
99
-
100
- with col2:
101
- st.markdown('Aurora GAN [model](https://huggingface.co/huggan/fastgan-few-shot-aurora-bs8)', unsafe_allow_html=True)
102
- st.image('aurora.png', width=220)
103
-
104
- with col3:
105
- st.markdown('Painting GAN [model](https://huggingface.co/huggan/fastgan-few-shot-painting-bs8)', unsafe_allow_html=True)
106
- st.image('painting.png', width=220)
107
- with col4:
108
- st.markdown('Shell GAN [model](https://huggingface.co/huggan/fastgan-few-shot-shells)', unsafe_allow_html=True)
109
- st.image('shell.png', width=220)
110
-
111
- st.markdown('___')
112
- if st.checkbox('Click if you want to create one of your own !'):
113
-
114
- col11, col12, col13 = st.columns([3,3.5,3.5])
115
  with col11:
116
- img_type = st.selectbox("Choose type of image to generate", index=0, options=["aurora", "painting", "fauvism", "shell"])
 
117
  # with col12:
118
  number_imgs = st.slider('How many images you want to generate ?', min_value=1, max_value=5)
119
  if number_imgs is None:
120
  st.write('Invalid number ! Please insert number of images to generate !')
121
  raise ValueError('Invalid number ! Please insert number of images to generate !')
122
 
123
- generate_button = st.button('Get Image!')
124
  if generate_button:
125
  st.markdown("""
126
  <small><i>Predictions may take up to 1 minute under high load. Please stand by.</i></small>
 
18
  model_name = {
19
  "aurora": 'huggan/fastgan-few-shot-aurora',
20
  "painting": 'huggan/fastgan-few-shot-painting',
 
21
  "shell": 'huggan/fastgan-few-shot-shells',
22
  "fauvism": 'huggan/fastgan-few-shot-fauvism-still-life',
23
+ "universe": 'huggan/fastgan-few-shot-universe',
24
+ "grumpy cat": 'huggan/fastgan-few-shot-grumpy-cat',
25
+ "anime": 'huggan/fastgan-few-shot-anime-face',
26
  }
27
 
28
  #@st.cache(allow_output_mutation=True)
 
56
  return None
57
  return r.json()
58
 
59
+ def show_model_summary(expanded):
60
+ with st.expander("Model gallery", expanded=expanded):
61
+ col1, col2, col3, col4 = st.columns(4)
62
+ with col1:
63
+ st.markdown('Fauvism GAN [model](https://huggingface.co/huggan/fastgan-few-shot-fauvism-still-life)', unsafe_allow_html=True)
64
+ st.image('fauvism.png', width=200)
65
+
66
+ st.markdown('Painting GAN [model](https://huggingface.co/huggan/fastgan-few-shot-painting)', unsafe_allow_html=True)
67
+ st.image('painting.png', width=200)
68
+
69
+ with col2:
70
+ st.markdown('Aurora GAN [model](https://huggingface.co/huggan/fastgan-few-shot-aurora)', unsafe_allow_html=True)
71
+ st.image('aurora.png', width=200)
72
+
73
+ st.markdown('Universe GAN [model](https://huggingface.co/huggan/fastgan-few-shot-universe)', unsafe_allow_html=True)
74
+ st.image('universe.png', width=200)
75
+
76
+ with col3:
77
+ st.markdown('Anime GAN [model](https://huggingface.co/huggan/fastgan-few-shot-anime-face)', unsafe_allow_html=True)
78
+ st.image('anime.png', width=200)
79
+
80
+ st.markdown('Shell GAN [model](https://huggingface.co/huggan/fastgan-few-shot-shells)', unsafe_allow_html=True)
81
+ st.image('shell.png', width=200)
82
+
83
+ with col4:
84
+ st.markdown('Grumpy cat GAN [model](https://huggingface.co/huggan/fastgan-few-shot-grumpy-cat)', unsafe_allow_html=True)
85
+ st.image('grumpy_cat.png', width=200)
86
+
87
 
88
  def main():
89
 
 
121
  )
122
 
123
  st.header("Welcome to FastGAN")
124
+ checked = st.checkbox('Click here if you want to create one of your own !')
125
 
126
+ if not checked:
127
+ show_model_summary(True)
128
+ if checked:
129
+ col11, col12, col13 = st.columns([2.9,3.05,3.05])
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
130
  with col11:
131
+ img_type = st.selectbox("Choose type of image to generate", index=0,
132
+ options=["aurora", "anime", "painting", "fauvism", "shell", "universe", "grumpy cat"])
133
  # with col12:
134
  number_imgs = st.slider('How many images you want to generate ?', min_value=1, max_value=5)
135
  if number_imgs is None:
136
  st.write('Invalid number ! Please insert number of images to generate !')
137
  raise ValueError('Invalid number ! Please insert number of images to generate !')
138
 
139
+ generate_button = st.button('Get Image')
140
  if generate_button:
141
  st.markdown("""
142
  <small><i>Predictions may take up to 1 minute under high load. Please stand by.</i></small>