Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -42,7 +42,7 @@ def predict_fracture(image):
|
|
42 |
img_with_boxes.rectangle([xmin, ymin, xmax, ymax], outline=color, width=2)
|
43 |
img_with_boxes.text((xmin, ymin), f"Fracture: {score:.2f}", font_size=12, color=color)
|
44 |
|
45 |
-
return Image.fromarray((np.uint8(img_with_boxes)))
|
46 |
|
47 |
# Define the to_tensor function
|
48 |
def to_tensor(image):
|
|
|
42 |
img_with_boxes.rectangle([xmin, ymin, xmax, ymax], outline=color, width=2)
|
43 |
img_with_boxes.text((xmin, ymin), f"Fracture: {score:.2f}", font_size=12, color=color)
|
44 |
|
45 |
+
return Image.fromarray((np.uint8(img_with_boxes)))
|
46 |
|
47 |
# Define the to_tensor function
|
48 |
def to_tensor(image):
|