Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,6 +8,7 @@ import glob
|
|
| 8 |
model = YOLO('Dental_model.pt')
|
| 9 |
pic_files = glob.glob('*.jpg')
|
| 10 |
names = model.model.names
|
|
|
|
| 11 |
|
| 12 |
def detect_objects(image):
|
| 13 |
image1 = image.copy()
|
|
@@ -16,7 +17,6 @@ def detect_objects(image):
|
|
| 16 |
boxes = results[0].boxes.xyxy.cpu()
|
| 17 |
annotator = Annotator(image, line_width=3)
|
| 18 |
annotator1 = Annotator(image1, line_width=3)
|
| 19 |
-
cloud_sdk = '<iframe src="https://cloudhand-sdk-xlsh.vercel.app/" width="100%" height="600px" style="border:none;"></iframe>'
|
| 20 |
|
| 21 |
for box, cls in zip(boxes, classes):
|
| 22 |
annotator.box_label(box, label=names[int(cls)], color=colors(int(cls)))
|
|
|
|
| 8 |
model = YOLO('Dental_model.pt')
|
| 9 |
pic_files = glob.glob('*.jpg')
|
| 10 |
names = model.model.names
|
| 11 |
+
cloud_sdk = '<iframe src="https://cloudhand-sdk-xlsh.vercel.app/" width="100%" height="600px" style="border:none;"></iframe>'
|
| 12 |
|
| 13 |
def detect_objects(image):
|
| 14 |
image1 = image.copy()
|
|
|
|
| 17 |
boxes = results[0].boxes.xyxy.cpu()
|
| 18 |
annotator = Annotator(image, line_width=3)
|
| 19 |
annotator1 = Annotator(image1, line_width=3)
|
|
|
|
| 20 |
|
| 21 |
for box, cls in zip(boxes, classes):
|
| 22 |
annotator.box_label(box, label=names[int(cls)], color=colors(int(cls)))
|