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")
|