ysharma HF staff commited on
Commit
58c5ba4
1 Parent(s): def888f

create app.py

Browse files
Files changed (1) hide show
  1. app.py +41 -0
app.py ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import pandas as pd
2
+ import gradio as gr
3
+
4
+ df = pd.read_csv("liked_images.csv")
5
+ df['url'] = df['url'].apply(lambda x: '<a href= "' + str(x) + '" target="_blank"> <img src= "' + str(x) + '"/> </a>') #'<img src= "' + str(x) + '"/> </a>')
6
+ df['seed'] = df['seed'].apply(lambda x: str(x))
7
+ df['width'] = df['width'].apply(lambda x: str(x))
8
+ df['height'] = df['height'].apply(lambda x: str(x))
9
+ df['steps'] = df['steps'].apply(lambda x: str(x))
10
+ df['source'] = df['source'].apply(lambda x: str(x))
11
+ df = df[[ 'url', 'prompt', 'seed', 'width', 'height', 'steps', 'source']]
12
+
13
+ def display_df():
14
+ df_images = df.head()
15
+ return df_images
16
+
17
+ def display_next10(dataframe, end):
18
+ start = (end or dataframe.index[-1]) + 1
19
+ end = start + 9
20
+ df_images = df.loc[start:end]
21
+ return df_images, end
22
+
23
+ #Gradio Blocks
24
+ with gr.Blocks() as demo:
25
+ gr.Markdown("<h1><center>Utility Gradio Space for viewing PlaygroundAI Images</center></h1>")
26
+ #gr.Markdown("""<img src='https://xxxxxxxx.jpg' class='center'> <br> """)
27
+ gr.Markdown(
28
+ """<div align="center">This Tool helps you to analyze and inspect the images and corresponding prompts from <a href = "https://playgroundai.com/">Playground AI</a> Images.<br><a href="https://twitter.com/Suhail">Suhail</a> has recently shared an open dataset of all the liked images and their prompts from PlaygroundAI on <a href="https://github.com/playgroundai/liked_images">Github here</a>. This is an attempt to explore this dataset beautifully using the power and flexibility of Gradio!<br><b>To use the tool:<br>First, click on the 'Initial' button, and then iteratively on the 'Next 10' button.<br><b>Bonus:</b>Click on images to get the original PlaygroundAI image displayed on next tab</div>""")
29
+
30
+ with gr.Row():
31
+ num_end = gr.Number(visible=False)
32
+ b1 = gr.Button("Get Initial dataframe")
33
+ b2 = gr.Button("Next 10 Rows")
34
+
35
+ with gr.Row():
36
+ out_dataframe = gr.Dataframe(wrap=True, max_rows=10, overflow_row_behaviour= "paginate", datatype = ["markdown", "markdown", "str", "str", "str", "str", "str", "str"])
37
+
38
+ b1.click(fn=display_df, outputs=out_dataframe)
39
+ b2.click(fn=display_next10, inputs= [out_dataframe, num_end ], outputs=[out_dataframe, num_end])
40
+
41
+ demo.launch(debug=True, show_error=True)