created demo options

app.py

@@ -10,47 +10,35 @@ from huggingface_hub import list_models
 
 from diffusers import StableDiffusionPipeline
 
-#submissions_list = list_models(filter=["dreambooth-hackathon", category], full=True)
-#spaces_pipeline_load = [submission.id for submission in submissions_list ]
-#for ids in spaces_pipeline_load:
-# mydict[ids] = StableDiffusionPipeline.from_pretrained(ids) #('ashiqabdulkhader/shiba-dog')
- #f"pipeline{ids.split('//')[-1]}" = StableDiffusionPipeline.from_pretrained(ids) #('ashiqabdulkhader/shiba-dog')
-
-#pipeline = StableDiffusionPipeline.from_pretrained("ashiqabdulkhader/shiba-dog") #('pharma/sugar-glider')
-#image = pipeline().images[0]
-#image
-#https://huggingface.co/ashiqabdulkhader/shiba-dog
-def filter_species(species):
- return gr.Dropdown.update(choices=species_map[species], value=species_map[species][1]), gr.update(visible=True)
-
- # %% app.ipynb 1
-def make_clickable_demo(model_name, prompt): #link=None):
- #if link is None:
- # link = "https://huggingface.co/" + model_name
- # Remove user from model name
- prompt = "a photo of " + ' '.join(model_name.split('/')[-1].split['-']) + str(prompt)
- 
- return gr.Button.update()
- 
-def make_clickable_model(model_name,prompt, link=None):
- if link is None:
- link = "https://huggingface.co/" + model_name
+# %% app.ipynb 1
+def get_model_list(category):
+ submissions_list = list_models(filter=["dreambooth-hackathon", category], full=True)
+ spaces_pipeline_load = [submission.id for submission in submissions_list ]
+ return gr.Dropdown.update(choices=spaces_pipeline_load , value=spaces_pipeline_load[5])
+
 
- #adding functionality for demo
- user_model_nm = model_name.split('/')[-1]
- print(f"user and model name is - {user_model_nm}")
+def get_initial_prompt(model_nm):
+ #a photo of peterj/shbrcky-dog 
+ user_model_nm = model_nm.split('/')[-1]
  if '-' in user_model_nm:
- model_nm = " ".join(user_model_nm.split('-'))
+ prompt = " ".join(user_model_nm.split('-'))
  else:
- model_nm = user_model_nm
- print(f"model name is - {model_nm}")
- prompt = "a photo of " + model_nm + str(prompt)
+ prompt = user_model_nm
+ return gr.Textbox.update(value="a photo of " + prompt + " ")
+
+def make_demo(model_name, prompt): #link=None):
+ #prompt = "a photo of " + ' '.join(model_name.split('/')[-1].split['-']) + str(prompt)
  pipeline = StableDiffusionPipeline.from_pretrained(model_name) #("ashiqabdulkhader/shiba-dog") #('pharma/sugar-glider')
- image_demo = pipeline(prompt).images[0]
- 
- # Remove user from model name
- return image_out.Update(value=image_demo, label=model_name.split("/")[-1]) #f'<a target="_blank" href="{link}">{model_name.split("/")[-1]}</a>'
+ image_demo = pipeline(prompt).images[0] 
+ return image_demo #gr.Button.update()
+
 
+def make_clickable_model(model_name, link=None):
+ if link is None:
+ link = "https://huggingface.co/" + model_name
+ # Remove user from model name
+ return f'<a target="_blank" href="{link}">{model_name.split("/")[-1]}</a>'
+ 
 def make_clickable_user(user_id):
  link = "https://huggingface.co/" + user_id
  return f'<a target="_blank" href="{link}">{user_id}</a>'
@@ -63,15 +51,20 @@ def get_submissions(category, prompt):
  for submission in submissions:
  # user, model, likes
  user_id = submission.id.split("/")[0]
+ model_nm = submission.id.split("/")[-1]
+ if '-' in model_nm:
+ model_nm = " ".join(model_nm.split('-'))
+ #button_html = get_button() 
  leaderboard_models.append(
  (
  make_clickable_user(user_id),
  make_clickable_model(submission.id, prompt),
  submission.likes,
+ #button_html #'a photo of ' + model_nm + " "
  )
  )
 
- df = pd.DataFrame(data=leaderboard_models, columns=["User", "Model", "Likes"])
+ df = pd.DataFrame(data=leaderboard_models, columns=["User", "Model", "Likes", ])
  df.sort_values(by=["Likes"], ascending=False, inplace=True)
  df.insert(0, "Rank", list(range(1, len(df) + 1)))
  return df
@@ -92,12 +85,15 @@ with block:
  )
  with gr.Row():
  prompt_in = gr.Textbox(label="Type in a Prompt. This will be suffixed to 'a photo of <model name>', so prompt accordingly -", value="")
- #button_in = gr.Button(label = "Generate Image using this model")
+ with gr.Column():
+ gr.Dropdown(label="Choose a dreamboooth model", value)
+ button_in = gr.Button(label = "Generate Image using this model")
+ image_out = gr.Image()
  with gr.Tabs():
  with gr.TabItem("Animal 🐨"):
  with gr.Row():
  animal_data = gr.components.Dataframe(
- type="pandas", datatype=["number", "markdown", "markdown", "number"]
+ type="pandas", datatype=["number", "markdown", "markdown", "number","str"], interactive = True
  )
  with gr.Row():
  data_run = gr.Button("Refresh")
@@ -109,7 +105,7 @@ with block:
  with gr.TabItem("Science 🔬"):
  with gr.Row():
  science_data = gr.components.Dataframe(
- type="pandas", datatype=["number", "markdown", "markdown", "number"]
+ type="pandas", datatype=["number", "markdown", "markdown", "number", "str"], interactive = True
  )
  with gr.Row():
  data_run = gr.Button("Refresh")
@@ -119,7 +115,7 @@ with block:
  with gr.TabItem("Food 🍔"):
  with gr.Row():
  food_data = gr.components.Dataframe(
- type="pandas", datatype=["number", "markdown", "markdown", "number"]
+ type="pandas", datatype=["number", "markdown", "markdown", "number", "str"], interactive = True
  )
  with gr.Row():
  data_run = gr.Button("Refresh")
@@ -129,7 +125,7 @@ with block:
  with gr.TabItem("Landscape 🏔"):
  with gr.Row():
  landscape_data = gr.components.Dataframe(
- type="pandas", datatype=["number", "markdown", "markdown", "number"]
+ type="pandas", datatype=["number", "markdown", "markdown", "number", "str"], interactive = True
  )
  with gr.Row():
  data_run = gr.Button("Refresh")
@@ -141,7 +137,7 @@ with block:
  with gr.TabItem("Wilcard 🔥"):
  with gr.Row():
  wildcard_data = gr.components.Dataframe(
- type="pandas", datatype=["number", "markdown", "markdown", "number"]
+ type="pandas", datatype=["number", "markdown", "markdown", "number", "str"], interactive = True
  )
  with gr.Row():
  data_run = gr.Button("Refresh")
@@ -151,11 +147,10 @@ with block:
  outputs=wildcard_data,
  )
  
- with gr.Row() : #as your_model_demo :
- image_out = gr.Image()
- 
- #button_in.click(make_clickable_demo, prompt_in, your_model_demo)
-
+ theme.change(get_model_list, theme, model_list ) 
+ model_list.change(get_initial_prompt, model_list, prompt_in )
+ button_in.click(make_demo, [model_list, prompt_in], image_out)
+ 
  block.load(get_submissions, inputs=[gr.Variable("animal"), prompt_in], outputs=animal_data)
  block.load(get_submissions, inputs=[gr.Variable("science"), prompt_in], outputs=science_data)
  block.load(get_submissions, inputs=[gr.Variable("food"), prompt_in], outputs=food_data)