Update app.py

app.py

@@ -74,10 +74,9 @@ embeddings, classes = load_embedding(model)
 
 iface = gr.Interface(fn=generate_image,
                      inputs=gr.Textbox(lines=2, placeholder="Enter Caption Here..."),
-                     outputs=gr.Image(type="numpy"),
+                     outputs=gr.Image(type="numpy", height=256, width=256),
                      title="CUHK Shenzhen Building Text-to-Image Generation",
-                     description="Enter a caption of some specific building in CUHK-Shenzhen to generate an image..",
-                     css=".output-image, .image-preview {height: 128px !important}"
+                     description="Enter a caption of some specific building in CUHK-Shenzhen to generate an image.."
                     )
 
 iface.launch(debug=True)