Size fix

app.py

@@ -1,5 +1,3 @@
-
-
 DEVICE = 'cpu'
 
 import gradio as gr
@@ -44,6 +42,7 @@ output_hidden_state = False
 def predict(
         prompt,
         im_emb=None,
+        progress=gr.Progress(track_tqdm=True)
     ):
     """Run a single prediction on the model"""
     with torch.no_grad():
@@ -158,8 +157,9 @@ def choose(choice, embs, ys, calibrate_prompts):
     img, embs, ys, calibrate_prompts = next_image(embs, ys, calibrate_prompts)
     return img, embs, ys, calibrate_prompts
 
-#css = "div#output-image {height: 768px !important; width: 768px !important; margin:auto; max-width: 100% !important;}"
-with gr.Blocks() as demo:
+css = ".gradio-container{max-width: 700px !important}"
+print(css)
+with gr.Blocks(css=css) as demo:
     embs = gr.State([])
     ys = gr.State([])
     calibrate_prompts = gr.State([
@@ -201,6 +201,6 @@ with gr.Blocks() as demo:
                  [b4, embs, ys, calibrate_prompts],
                  [b1, b2, b3, b4, img, embs, ys, calibrate_prompts])
     with gr.Row():
-        html = gr.HTML('''<div style='text-align:center; font-size:32'>You will calibrate for several prompts and then roam.</ div>''')
+        html = gr.HTML('''<div style='text-align:center; font-size:20px'>You will calibrate for several prompts and then roam.</ div>''')
 
 demo.launch()  # Share your demo with just 1 extra parameter 🚀