Ricercar commited on
Commit
d063f18
โ€ข
1 Parent(s): d504e45

add descriptions to homepage

Browse files
Files changed (1) hide show
  1. Home.py +33 -6
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 for personalization ๐Ÿ‘‡",
15
- label_visibility='visible',
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 ai generated images. \
49
- You can know which model you like most after you finish the survey."
50
  )
51
 
52
  if 'user_id' not in st.session_state:
53
  login()
54
  else:
55
- st.write('You have already logged in as ' + st.session_state.user_id[0])
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')