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