Spaces:
Running
Running
Update app.py
Browse files
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 |
-
|
|
|
|
|
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 |
-
|
|
|
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
|