fix

app.py

@@ -181,7 +181,7 @@ def load_model(dataset, extra_kwargs={}, device='cuda'):
 
 device = 'cuda' if th.cuda.is_available() else 'cpu'
 
-clevr_model = load_model('clevr', extra_kwargs=dict(embed_dim=64, enc_channels=128), device=device)
+clevr_model = load_model('clevr', extra_kwargs=dict(emb_dim=64, enc_channels=128), device=device)
 celeb_model = load_model('celebahq', extra_kwargs=dict(enc_channels=128), device=device)
 
 MODELS = {
@@ -200,7 +200,7 @@ with gr.Blocks() as demo:
 </p>""")
     
     gr.Markdown(
-            """<h4>Decomposition and reconstruction of images</h4>""")
+            """<br> <h4>Decomposition and reconstruction of images</h4>""")
     with gr.Row():
         with gr.Column():
             with gr.Row():
@@ -227,7 +227,7 @@ with gr.Blocks() as demo:
 
 
     gr.Markdown(
-            """<h4>Combination of images</h4>""")
+            """<br> <h4>Combination of images</h4>""")
     with gr.Row().style(equal_height=True):
         with gr.Column(scale=2):