BioMike commited on
Commit
add62f2
1 Parent(s): eae2fb2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -19
app.py CHANGED
@@ -1,20 +1,20 @@
1
- import gradio as gr
2
- from vae import vae
3
- from morphing import morphing
4
- from model import ConvVAE
5
-
6
- model = ConvVAE.from_pretrained(
7
- model_id="BioMike/classical_portrait_vae",
8
- cache_dir="./model_cache",
9
- map_location="cpu",
10
- strict=True
11
- ).eval()
12
-
13
-
14
- demo = gr.TabbedInterface([vae, morphing],
15
- ["Image to Portrait", "Image to Image (Morphing)"]
16
- title="CLassical Portraits VAE",
17
- theme=gr.themes.Base())
18
-
19
- demo.queue()
20
  demo.launch(debug=True, share=True)
 
1
+ import gradio as gr
2
+ from vae import vae
3
+ from morphing import morphing
4
+ from model import ConvVAE
5
+
6
+ model = ConvVAE.from_pretrained(
7
+ model_id="BioMike/classical_portrait_vae",
8
+ cache_dir="./model_cache",
9
+ map_location="cpu",
10
+ strict=True
11
+ ).eval()
12
+
13
+
14
+ demo = gr.TabbedInterface([vae, morphing],
15
+ ["Image to Portrait", "Image to Image (Morphing)"],
16
+ title="CLassical Portraits VAE",
17
+ theme=gr.themes.Base())
18
+
19
+ demo.queue()
20
  demo.launch(debug=True, share=True)