Spaces:

AIEnergyScore
/

Label

Running

bgamazay commited on Jan 26

Commit

0b88af7

verified ·

1 Parent(s): ea74654

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -54,7 +54,7 @@ def main():
     final_size = (520, 728)
     generated_label = create_label_single_pass(background, model_data, final_size)
-    st.image(generated_label, caption="Generated Label Preview", use_column_width=False)
     img_buffer = io.BytesIO()
     generated_label.save(img_buffer, format="PNG")

     final_size = (520, 728)
     generated_label = create_label_single_pass(background, model_data, final_size)
+    st.image(generated_label, caption="Generated Label Preview", width=520)
     img_buffer = io.BytesIO()
     generated_label.save(img_buffer, format="PNG")