bgamazay commited on
Commit
014c338
·
verified ·
1 Parent(s): 1867c6b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -6
app.py CHANGED
@@ -79,7 +79,6 @@ def main():
79
  # Task selection
80
  st.sidebar.write("#### 1. Select task(s) to view models")
81
  selected_tasks = st.sidebar.multiselect("", options=task_order, default=task_order)
82
-
83
  # Mapping from task to CSV file name.
84
  task_to_file = {
85
  "Text Generation": "text_gen_energyscore.csv",
@@ -93,10 +92,7 @@ def main():
93
  "Question Answering": "question_answering_energyscore.csv",
94
  "Sentence Similarity": "sentence_similarity_energyscore.csv"
95
  }
96
-
97
  st.sidebar.write("#### 2. Select a model to generate label")
98
-
99
-
100
  default_model_data = {
101
  'provider': "AI Provider",
102
  'model': "Model Name",
@@ -185,7 +181,6 @@ def main():
185
  st.sidebar.write("- [FAQ](https://huggingface.github.io/AIEnergyScore/#faq)")
186
  st.sidebar.write("- [Documentation](https://huggingface.github.io/AIEnergyScore/#documentation)")
187
 
188
-
189
  def create_label_single_pass(background_image, model_data, final_size=(520, 728)):
190
  bg_resized = background_image.resize(final_size, Image.Resampling.LANCZOS)
191
  draw = ImageDraw.Draw(bg_resized)
@@ -200,7 +195,7 @@ def create_label_single_pass(background_image, model_data, final_size=(520, 728)
200
 
201
  title_x, title_y = 33, 150
202
  details_x, details_y = 480, 256
203
- energy_x, energy_y = 440, 472
204
 
205
  draw.text((title_x, title_y), str(model_data['provider']), font=title_font, fill="black")
206
  draw.text((title_x, title_y + 38), str(model_data['model']), font=title_font, fill="black")
 
79
  # Task selection
80
  st.sidebar.write("#### 1. Select task(s) to view models")
81
  selected_tasks = st.sidebar.multiselect("", options=task_order, default=task_order)
 
82
  # Mapping from task to CSV file name.
83
  task_to_file = {
84
  "Text Generation": "text_gen_energyscore.csv",
 
92
  "Question Answering": "question_answering_energyscore.csv",
93
  "Sentence Similarity": "sentence_similarity_energyscore.csv"
94
  }
 
95
  st.sidebar.write("#### 2. Select a model to generate label")
 
 
96
  default_model_data = {
97
  'provider': "AI Provider",
98
  'model': "Model Name",
 
181
  st.sidebar.write("- [FAQ](https://huggingface.github.io/AIEnergyScore/#faq)")
182
  st.sidebar.write("- [Documentation](https://huggingface.github.io/AIEnergyScore/#documentation)")
183
 
 
184
  def create_label_single_pass(background_image, model_data, final_size=(520, 728)):
185
  bg_resized = background_image.resize(final_size, Image.Resampling.LANCZOS)
186
  draw = ImageDraw.Draw(bg_resized)
 
195
 
196
  title_x, title_y = 33, 150
197
  details_x, details_y = 480, 256
198
+ energy_x, energy_y = 430, 472
199
 
200
  draw.text((title_x, title_y), str(model_data['provider']), font=title_font, fill="black")
201
  draw.text((title_x, title_y + 38), str(model_data['model']), font=title_font, fill="black")