feat: info, fix mkdir error, bottom download button

app.py

@@ -18,6 +18,14 @@ def hex_to_rgb(s: str) -> tuple[int, int, int]:
     return (int(value[:2], 16), int(value[2:4], 16), int(value[4:6], 16))
 
 
+st.header("Color Textual Inversion")
+with st.expander(label="info"):
+    with open("info.txt", "r", encoding="utf-8") as f:
+        st.markdown(f.read())
+
+duplicate_button = """<a class="duplicate-button" style="display:inline-block" target="_blank" href="https://huggingface.co/spaces/Bingsu/color_textual_inversion?duplicate=true"><img style="margin: 0" src="https://img.shields.io/badge/-Duplicate%20Space-blue?labelColor=white&style=flat&logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAYAAAAf8/9hAAAAAXNSR0IArs4c6QAAAP5JREFUOE+lk7FqAkEURY+ltunEgFXS2sZGIbXfEPdLlnxJyDdYB62sbbUKpLbVNhyYFzbrrA74YJlh9r079973psed0cvUD4A+4HoCjsA85X0Dfn/RBLBgBDxnQPfAEJgBY+A9gALA4tcbamSzS4xq4FOQAJgCDwV2CPKV8tZAJcAjMMkUe1vX+U+SMhfAJEHasQIWmXNN3abzDwHUrgcRGmYcgKe0bxrblHEB4E/pndMazNpSZGcsZdBlYJcEL9Afo75molJyM2FxmPgmgPqlWNLGfwZGG6UiyEvLzHYDmoPkDDiNm9JR9uboiONcBXrpY1qmgs21x1QwyZcpvxt9NS09PlsPAAAAAElFTkSuQmCC&logoWidth=14" alt="Duplicate Space"></a>"""
+st.markdown(duplicate_button, unsafe_allow_html=True)
+
 col1, col2 = st.columns([15, 85])
 color = col1.color_picker("Pick a color", "#00f900")
 col2.text_input("", color, disabled=True)
@@ -38,6 +46,7 @@ if output_path.exists():
     shutil.rmtree(output_path)
 
 dataset_path.mkdir()
+output_path.mkdir()
 img_path = dataset_path / f"{emb_name}.png"
 Image.fromarray(img_array).save(img_path)
 
@@ -115,3 +124,4 @@ torch.save(trained_emb, result_path)
 
 file = result_path.read_bytes()
 download_button.download_button(f"Download {emb_name}.pt", file, f"{emb_name}.pt")
+st.download_button(f"Download {emb_name}.pt ", file, f"{emb_name}.pt")

info.txt

@@ -0,0 +1,7 @@
+Create an embedding that represents a color code.
+
+Using only one simple color image, textual inversion training is performed.
+
+This idea is from the arcalive AI image channel, [내가 원하는 색상코드를 만들어 사용해 보자](https://arca.live/b/aiart/64702219).
+
+However, this space uses an implementation of huggingface diffusers, so the result is different from webui. Please be careful on this point.