svjack commited on
Commit
9f9f32c
·
1 Parent(s): 59c5d79

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ dataset = load_dataset("Gustavosta/Stable-Diffusion-Prompts")
11
  prompt_df = dataset["train"].to_pandas()
12
 
13
  def get_samples():
14
- prompt_list = prompt_df.sample(n = 10)["Prompt"].values.tolist()
15
  return prompt_list
16
 
17
  def get_params(request: gr.Request):
 
11
  prompt_df = dataset["train"].to_pandas()
12
 
13
  def get_samples():
14
+ prompt_list = prompt_df.sample(n = 10)["Prompt"].reset_index().values.tolist()
15
  return prompt_list
16
 
17
  def get_params(request: gr.Request):