Spaces:
Running
Running
add descriptions to homepage
Browse files
Home.py
CHANGED
@@ -11,8 +11,8 @@ def login():
|
|
11 |
with st.form("user_login"):
|
12 |
st.write('## Enter Your Name')
|
13 |
user_id = st.text_input(
|
14 |
-
"Enter your name
|
15 |
-
label_visibility='
|
16 |
disabled=False,
|
17 |
placeholder='anonymous',
|
18 |
)
|
@@ -43,15 +43,42 @@ if __name__ == '__main__':
|
|
43 |
|
44 |
st.set_page_config(page_title="Login", page_icon="🏠", layout="wide")
|
45 |
st.write('A Research by MAPS Lab, NYU Shanghai')
|
46 |
-
st.title("Welcome to GEMRec Gallery Webapp!")
|
47 |
st.write(
|
48 |
-
"This is an web application to collect personal preference to
|
49 |
-
You
|
50 |
)
|
51 |
|
52 |
if 'user_id' not in st.session_state:
|
53 |
login()
|
54 |
else:
|
55 |
-
st.write(
|
56 |
st.button('Log out', on_click=logout)
|
57 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
11 |
with st.form("user_login"):
|
12 |
st.write('## Enter Your Name')
|
13 |
user_id = st.text_input(
|
14 |
+
"Enter your name 👇",
|
15 |
+
label_visibility='collapsed',
|
16 |
disabled=False,
|
17 |
placeholder='anonymous',
|
18 |
)
|
|
|
43 |
|
44 |
st.set_page_config(page_title="Login", page_icon="🏠", layout="wide")
|
45 |
st.write('A Research by MAPS Lab, NYU Shanghai')
|
46 |
+
st.title("🙌 Welcome to GEMRec Gallery Webapp!")
|
47 |
st.write(
|
48 |
+
"This is an web application to collect personal preference to images synthesised by generative models fine-tuned on stable diffusion. \
|
49 |
+
**You might consider it as a tool for quickly digging out the most suitable model for you from [civitai](https://civitai.com/).**"
|
50 |
)
|
51 |
|
52 |
if 'user_id' not in st.session_state:
|
53 |
login()
|
54 |
else:
|
55 |
+
st.write(f"You have already logged in as `{st.session_state.user_id[0]}`")
|
56 |
st.button('Log out', on_click=logout)
|
57 |
|
58 |
+
st.write('---')
|
59 |
+
st.write('## FAQ')
|
60 |
+
with st.expander(label='**🤔 How to use this webapp?**'):
|
61 |
+
st.write('### Check out the demo video below')
|
62 |
+
st.video('https://youtu.be/EjjCoeUSZF0')
|
63 |
+
st.caption('Interface shown in this video demo might be a bit different from the current webapp because of further updates, but the basic idea is the same.')
|
64 |
+
|
65 |
+
with st.expander(label='**ℹ️ What is GEMRec project?**'):
|
66 |
+
st.write('### About GEMRec')
|
67 |
+
st.write("**GE**nerative **M**odel **Rec**ommendation (**GEMRec**) is a research project by [MAPS Lab](https://github.com/MAPS-research), NYU Shanghai.")
|
68 |
+
st.write('### Our Task')
|
69 |
+
st.write('Given a user’s preference on a set of generated images, we aim to recommend the most preferred generative model for the user.')
|
70 |
+
st.write('### Our Approach')
|
71 |
+
st.write('We propose a two-stage framework, which contains prompt-model retrival and generated item ranking. :red[Your participation in this web application will help us to improve our framework and to further our research on personalization.]')
|
72 |
+
st.write('### Key Contributions')
|
73 |
+
st.write('1. We propose a two-stage framework to approach the Generative Model Recommendation problem. Our framework allows end-users to effectively explore a diverse set of generative models to understand their expressiveness. It also allows system developers to elicit user preferences for items generated from personalized prompts.')
|
74 |
+
st.write('2. We release GEMRec-18K, a dense prompt-model interaction dataset that consists of 18K images generated by pairing 200 generative models with 90 prompts collected from real-world usages, accompanied by detailed metadata and generation configurations. This dataset builds the cornerstone for exploring Generative Recommendation and can be useful for other tasks related to understanding generative models')
|
75 |
+
st.write('3. We take the first step in examining evaluation metrics for personalized image generations and identify several limitations in existing metrics. We propose a weighted metric that is more suitable for the task and opens up directions for future improvements in model training and evaluations.')
|
76 |
+
|
77 |
+
with st.expander(label='**💻 Where can I find the paper and dataset?**'):
|
78 |
+
st.write('### Paper')
|
79 |
+
st.write('Arxiv: [Towards Personalized Prompt-Model Retrieval for Generative Recommendation](https://arxiv.org/abs/2308.02205)')
|
80 |
+
st.write('### GEMRec-18K Dataset')
|
81 |
+
st.write('Image dataset: https://huggingface.co/datasets/MAPS-research/GEMRec-PromptBook \n \
|
82 |
+
Model dataset: https://huggingface.co/datasets/MAPS-research/GEMRec-Roster')
|
83 |
+
st.write('### Code')
|
84 |
+
st.write('Github: https://github.com/maps-research/gemrec')
|