davidberenstein1957 commited on
Commit
a65206d
Β·
1 Parent(s): 0c350fd

refactor: streamline DataFrame column selection in app.py, update logo image for branding consistency, and enhance "About" section with additional links for improved user engagement

Browse files
Files changed (1) hide show
  1. app.py +13 -16
app.py CHANGED
@@ -24,11 +24,7 @@ df["Model"] = df.apply(
24
  )
25
  df = df[
26
  ["Model", "Median Inference Time", "Price per Image"]
27
- + [
28
- col
29
- for col in df.columns.tolist()
30
- if col not in ["URL", "Model", "Median Inference Time", "Price per Image"]
31
- ]
32
  ]
33
  df = df.sort_values(by="GenEval", ascending=False)
34
 
@@ -36,7 +32,7 @@ with gr.Blocks("ParityError/Interstellar", css=custom_css) as demo:
36
  gr.HTML(
37
  """
38
  <div style="text-align: center;">
39
- <img src="https://huggingface.co/datasets/PrunaAI/documentation-images/resolve/main/inferbench/logo1.png" style="width: 200px; height: auto; max-width: 100%; margin: 0 auto;">
40
  <h1>πŸ‹οΈ InferBench πŸ‹οΈ</h1>
41
  <h2>A cost/quality/speed Leaderboard for Inference Providers!</h2>
42
  </div>
@@ -45,12 +41,8 @@ with gr.Blocks("ParityError/Interstellar", css=custom_css) as demo:
45
 
46
  with gr.Tabs():
47
  with gr.TabItem("FLUX.1 [dev] Leaderboard"):
48
- median_inference_time_min = math.floor(
49
- float(df["Median Inference Time"].min())
50
- )
51
- median_inference_time_max = math.ceil(
52
- float(df["Median Inference Time"].max())
53
- )
54
  price_per_image_min = math.floor(float(df["Price per Image"].min()))
55
  price_per_image_max = math.ceil(float(df["Price per Image"].max()))
56
  Leaderboard(
@@ -81,7 +73,7 @@ with gr.Blocks("ParityError/Interstellar", css=custom_css) as demo:
81
  gr.Markdown(
82
  """
83
  # πŸ’œ About Pruna AI
84
- We are Pruna AI, an open source AI optimisation engine and we simply make your models cheaper, faster, smaller, greener!
85
 
86
  # πŸ“Š About InferBench
87
  InferBench is a leaderboard for inference providers, focusing on cost, quality, and speed.
@@ -97,11 +89,16 @@ with gr.Blocks("ParityError/Interstellar", css=custom_css) as demo:
97
  FLUX-juiced was obtained using a combination of compilation and caching algorithms and we are proud to say that it consistently outperforms alternatives, while delivering performance on par with the original model.
98
  This combination is available in our Pruna Pro package and can be applied to almost every image generation model.
99
 
100
- A full blogpost on the method can be found [here](https://pruna.ai/blog/flux-juiced). # TODO: Add link
 
101
  """
102
  )
103
  with gr.Column(scale=1):
104
- gr.Markdown("")
 
 
 
 
105
 
106
  with gr.Accordion("🌍 Join the Pruna AI community!", open=False):
107
  gr.HTML(
@@ -111,7 +108,7 @@ with gr.Blocks("ParityError/Interstellar", css=custom_css) as demo:
111
  <a rel="nofollow" href="https://www.linkedin.com/company/93832878/admin/feed/posts/?feedType=following"><img alt="LinkedIn" src="https://img.shields.io/badge/LinkedIn-Connect-blue"></a>
112
  <a rel="nofollow" href="https://discord.com/invite/rskEr4BZJx"><img alt="Discord" src="https://img.shields.io/badge/Discord-Join%20Us-blue?style=social&amp;logo=discord"></a>
113
  <a rel="nofollow" href="https://www.reddit.com/r/PrunaAI/"><img alt="Reddit" src="https://img.shields.io/reddit/subreddit-subscribers/PrunaAI?style=social"></a>
114
- """
115
  )
116
  with gr.Accordion("Citation", open=True):
117
  gr.Markdown(
 
24
  )
25
  df = df[
26
  ["Model", "Median Inference Time", "Price per Image"]
27
+ + [col for col in df.columns.tolist() if col not in ["URL", "Model", "Median Inference Time", "Price per Image"]]
 
 
 
 
28
  ]
29
  df = df.sort_values(by="GenEval", ascending=False)
30
 
 
32
  gr.HTML(
33
  """
34
  <div style="text-align: center;">
35
+ <img src="https://huggingface.co/datasets/PrunaAI/documentation-images/resolve/main/inferbench/logo2-cropped.png" style="width: 200px; height: auto; max-width: 100%; margin: 0 auto;">
36
  <h1>πŸ‹οΈ InferBench πŸ‹οΈ</h1>
37
  <h2>A cost/quality/speed Leaderboard for Inference Providers!</h2>
38
  </div>
 
41
 
42
  with gr.Tabs():
43
  with gr.TabItem("FLUX.1 [dev] Leaderboard"):
44
+ median_inference_time_min = math.floor(float(df["Median Inference Time"].min()))
45
+ median_inference_time_max = math.ceil(float(df["Median Inference Time"].max()))
 
 
 
 
46
  price_per_image_min = math.floor(float(df["Price per Image"].min()))
47
  price_per_image_max = math.ceil(float(df["Price per Image"].max()))
48
  Leaderboard(
 
73
  gr.Markdown(
74
  """
75
  # πŸ’œ About Pruna AI
76
+ We are [Pruna AI, an open source AI optimisation engine](https://github.com/PrunaAI/pruna) and we simply make your models cheaper, faster, smaller, greener!
77
 
78
  # πŸ“Š About InferBench
79
  InferBench is a leaderboard for inference providers, focusing on cost, quality, and speed.
 
89
  FLUX-juiced was obtained using a combination of compilation and caching algorithms and we are proud to say that it consistently outperforms alternatives, while delivering performance on par with the original model.
90
  This combination is available in our Pruna Pro package and can be applied to almost every image generation model.
91
 
92
+ - A full blogpost on the methodology can be found [here](https://pruna.ai/blog/flux-juiced).
93
+ - A website that compares the outputs of the different models can be found [here](https://www.notion.so/FLUX-juiced-1d270a039e5f80c6a2a3c00fc0d75ef0?pvs=4).
94
  """
95
  )
96
  with gr.Column(scale=1):
97
+ gr.HTML(
98
+ """
99
+ <iframe src="https://www.notion.so/FLUX-juiced-1d270a039e5f80c6a2a3c00fc0d75ef0?pvs=4" width="100%" height="100%" frameborder="0"></iframe>
100
+ """
101
+ )
102
 
103
  with gr.Accordion("🌍 Join the Pruna AI community!", open=False):
104
  gr.HTML(
 
108
  <a rel="nofollow" href="https://www.linkedin.com/company/93832878/admin/feed/posts/?feedType=following"><img alt="LinkedIn" src="https://img.shields.io/badge/LinkedIn-Connect-blue"></a>
109
  <a rel="nofollow" href="https://discord.com/invite/rskEr4BZJx"><img alt="Discord" src="https://img.shields.io/badge/Discord-Join%20Us-blue?style=social&amp;logo=discord"></a>
110
  <a rel="nofollow" href="https://www.reddit.com/r/PrunaAI/"><img alt="Reddit" src="https://img.shields.io/reddit/subreddit-subscribers/PrunaAI?style=social"></a>
111
+ """
112
  )
113
  with gr.Accordion("Citation", open=True):
114
  gr.Markdown(