Update app.py

app.py

@@ -49,7 +49,7 @@ def init():
 
 img = mpimg.imread('figure.png')
 # Output Images
-st.image(img, width=1000, use_column_width='always')
+st.image(img, width=1000)
 model, text_tokenizer, cond_tokenizer, args, trainer = init()