Update app.py

app.py

@@ -361,13 +361,27 @@ def img2img_tab():
 
 def main():
     st.title("FFusion AI -beta- Playground")
-
+    st.markdown("""
+    [![Hugging Face Model](https://img.shields.io/badge/Hugging%20Face-FFusion--BaSE-blue)](https://huggingface.co/FFusion/FFusion-BaSE)
+    [![GitHub](https://img.shields.io/badge/GitHub-1e--2-green)](https://github.com/1e-2)
+    [![Facebook](https://img.shields.io/badge/Facebook-FFusionAI-blue)](https://www.facebook.com/FFusionAI/)
+    [![Civitai](https://img.shields.io/badge/Civitai-FFusionAI-blue)](https://civitai.com/models/82039/ffusion-ai-sd-21)
+    [![DOI](https://img.shields.io/badge/DOI-10.57967%2Fhf%2F0851-red)](https://huggingface.co/models?other=doi%3A10.57967%2Fhf%2F0851)
+    """)
     tabs = ["FFusion BaSE 768+ (txt2img)"]
     selected_tab = st.selectbox("Choose a di.FFusion.ai model", tabs)
 
     if selected_tab == "FFusion BaSE 768+ (txt2img)":
         txt2img_tab()
 
+        
+    st.markdown("""
+    Models: [![FFusion-BaSE](https://img.shields.io/badge/2.1%20🤗%20Model-FFusion--BaSE-blue)](https://huggingface.co/FFusion/FFusion-BaSE)    [![di.FFUSION.ai-v2.1-768-BaSE-alpha](https://img.shields.io/badge/🤗%20Model-di.FFUSION.ai--v2.1--768--BaSE--alpha-blue)](https://huggingface.co/FFusion/di.FFUSION.ai-v2.1-768-BaSE-alpha)
+    [![di.ffusion.ai.Beta512](https://img.shields.io/badge/2.1%20🤗%20Model-di.ffusion.ai.Beta512-blue)](https://huggingface.co/FFusion/di.ffusion.ai.Beta512)    [![FFUSION.ai-Text-Encoder-LyCORIS-SD-2.1](https://img.shields.io/badge/2.1%20🤗%20Model-FFUSION.ai--Text--Encoder--LyCORIS--SD--2.1-blue)](https://huggingface.co/FFusion/FFUSION.ai-Text-Encoder-LyCORIS-SD-2.1)
+    Contact: [![Email](https://img.shields.io/badge/Email-di%40ffusion.ai-blue)](mailto:[email protected])
+    """)
+
+    
     st.header("Citation")
 
     """