bgamazay commited on
Commit
9e2fdc0
·
verified ·
1 Parent(s): 35c757c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -92,14 +92,14 @@ def create_label(background_image, model_data):
92
  energy_x, energy_y = 600, 600
93
 
94
  # Group 1: Title (Left-Justified, no prefixes)
95
- draw.text((title_x, title_y), model_data['model'], font=title_font, fill="black")
96
- draw.text((title_x, title_y + 25), model_data['provider'], font=title_font, fill="black")
97
 
98
  # Group 2: Details (Right-Justified, no prefixes)
99
  details_lines = [
100
- model_data['date'],
101
- model_data['task'],
102
- model_data['hardware']
103
  ]
104
  for i, line in enumerate(details_lines):
105
  bbox = draw.textbbox((0, 0), line, font=details_font)
@@ -107,7 +107,7 @@ def create_label(background_image, model_data):
107
  draw.text((details_x - text_width, details_y + i * 20), line, font=details_font, fill="black")
108
 
109
  # Group 3: Energy (Bottom-Center, no prefixes)
110
- energy_text = model_data['energy']
111
  bbox = draw.textbbox((0, 0), energy_text, font=energy_font)
112
  energy_text_width = bbox[2] - bbox[0]
113
  draw.text((energy_x - energy_text_width // 2, energy_y), energy_text, font=energy_font, fill="black")
 
92
  energy_x, energy_y = 600, 600
93
 
94
  # Group 1: Title (Left-Justified, no prefixes)
95
+ draw.text((title_x, title_y), str(model_data['model']), font=title_font, fill="black")
96
+ draw.text((title_x, title_y + 25), str(model_data['provider']), font=title_font, fill="black")
97
 
98
  # Group 2: Details (Right-Justified, no prefixes)
99
  details_lines = [
100
+ str(model_data['date']),
101
+ str(model_data['task']),
102
+ str(model_data['hardware'])
103
  ]
104
  for i, line in enumerate(details_lines):
105
  bbox = draw.textbbox((0, 0), line, font=details_font)
 
107
  draw.text((details_x - text_width, details_y + i * 20), line, font=details_font, fill="black")
108
 
109
  # Group 3: Energy (Bottom-Center, no prefixes)
110
+ energy_text = str(model_data['energy']) # Ensure this is a string
111
  bbox = draw.textbbox((0, 0), energy_text, font=energy_font)
112
  energy_text_width = bbox[2] - bbox[0]
113
  draw.text((energy_x - energy_text_width // 2, energy_y), energy_text, font=energy_font, fill="black")