bgamazay commited on
Commit
25ce5b4
·
verified ·
1 Parent(s): 47a0f82

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -96,12 +96,15 @@ def create_label(background_image, model_data):
96
  f"Hardware: {model_data['hardware']}"
97
  ]
98
  for i, line in enumerate(details_lines):
99
- text_width, _ = draw.textsize(line, font=details_font)
 
 
100
  draw.text((details_x - text_width, details_y + i * 20), line, font=details_font, fill="black")
101
 
102
  # Group 3: Energy (Bottom-Center)
103
  energy_text = f"Energy: {model_data['energy']}"
104
- energy_text_width, _ = draw.textsize(energy_text, font=energy_font)
 
105
  draw.text((energy_x - energy_text_width // 2, energy_y), energy_text, font=energy_font, fill="black")
106
 
107
  return label_img
 
96
  f"Hardware: {model_data['hardware']}"
97
  ]
98
  for i, line in enumerate(details_lines):
99
+ # Use textbbox to calculate text width
100
+ bbox = draw.textbbox((0, 0), line, font=details_font)
101
+ text_width = bbox[2] - bbox[0] # Right - Left
102
  draw.text((details_x - text_width, details_y + i * 20), line, font=details_font, fill="black")
103
 
104
  # Group 3: Energy (Bottom-Center)
105
  energy_text = f"Energy: {model_data['energy']}"
106
+ bbox = draw.textbbox((0, 0), energy_text, font=energy_font)
107
+ energy_text_width = bbox[2] - bbox[0]
108
  draw.text((energy_x - energy_text_width // 2, energy_y), energy_text, font=energy_font, fill="black")
109
 
110
  return label_img