barretto commited on
Commit
a3ed1cd
·
1 Parent(s): 09e9257

added latin ethnicity

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -24,15 +24,15 @@ def getImage(prompt):
24
 
25
  demo = gr.Interface(
26
  fn = run,
27
- article = "<p>For better accuracy, please enter the following information about the person you want to generate a portrait for.</p><h3>Biases and content acknowledgment</h3><p>Beware to the fact that this is a pre-trained model that may output content that reinforces or exacerbates societal biases, as well as realistic faces, pornography and violence. The model was trained on the LAION-5B dataset, which scraped non-curated image-text-pairs from the internet (the exception being the removal of illegal content) and is meant for research purposes.</p><p>You can read more in the model card <a href=\"https://huggingface.co/CompVis/stable-diffusion-v1-4\" target=\"_blank\">HERE.</a></p>"
28
  ,
29
  inputs = [
30
 
31
  gr.Radio(["Male", "Female"], label="Gender" ), #Gender
32
  gr.Slider(100,250,value=160, label="Height (cm)"), #Height
33
  gr.Slider(40,250,value=50, label="Weight (kg)"), #Weight
34
- gr.Radio(sorted(["Young", "Adult", "Middle-aged", "Old"]), label="Age" ), #Age
35
- gr.Dropdown(sorted(["South Asian", "North Asian", "White", "African American", "American Indian", "Hispanic"]), label="Ethnicity"), #Ethnicity
36
  gr.Dropdown(sorted(["Black", "Brown", "Yellow", "Peach","Tan","Beige", "White", "Grey"]), label="Skin color" ), #Skin color
37
  gr.Radio(sorted(["Short", "Long", "Bald", "Medium"]), label="Hair length" ), #Hair length
38
  gr.Dropdown(sorted(["Black","Dark brown", "Light brown", "Blond", "Red", "Grey"]), label="Hair color" ), #Hair color
 
24
 
25
  demo = gr.Interface(
26
  fn = run,
27
+ article = "<p>For better accuracy, please enter all the information about the person you want to generate a portrait for.</p><h3>Biases and content acknowledgment</h3><p>Beware to the fact that this is a pre-trained model that may output content that reinforces or exacerbates societal biases, as well as realistic faces, pornography and violence. The model was trained on the LAION-5B dataset, which scraped non-curated image-text-pairs from the internet (the exception being the removal of illegal content) and is meant for research purposes.</p><p>You can read more in the model card <a href=\"https://huggingface.co/CompVis/stable-diffusion-v1-4\" target=\"_blank\">HERE.</a></p>"
28
  ,
29
  inputs = [
30
 
31
  gr.Radio(["Male", "Female"], label="Gender" ), #Gender
32
  gr.Slider(100,250,value=160, label="Height (cm)"), #Height
33
  gr.Slider(40,250,value=50, label="Weight (kg)"), #Weight
34
+ gr.Radio(["Young", "Adult", "Middle-aged", "Old"], label="Age" ), #Age
35
+ gr.Dropdown(sorted(["South Asian", "North Asian", "White", "African American", "American Indian", "Hispanic", "Latin"]), label="Ethnicity"), #Ethnicity
36
  gr.Dropdown(sorted(["Black", "Brown", "Yellow", "Peach","Tan","Beige", "White", "Grey"]), label="Skin color" ), #Skin color
37
  gr.Radio(sorted(["Short", "Long", "Bald", "Medium"]), label="Hair length" ), #Hair length
38
  gr.Dropdown(sorted(["Black","Dark brown", "Light brown", "Blond", "Red", "Grey"]), label="Hair color" ), #Hair color