Spaces:
Runtime error
Runtime error
siddhantgore
commited on
Commit
•
a499985
1
Parent(s):
4b0a60e
Update app.py
Browse files
app.py
CHANGED
@@ -54,7 +54,7 @@ def inference(image):
|
|
54 |
#img = np.array(image.resize((800, height)))
|
55 |
img = np.array(image)
|
56 |
outputs = predictor(img)
|
57 |
-
v=Visualizer(
|
58 |
out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
|
59 |
return out.get_image()
|
60 |
if file is None:
|
|
|
54 |
#img = np.array(image.resize((800, height)))
|
55 |
img = np.array(image)
|
56 |
outputs = predictor(img)
|
57 |
+
v=Visualizer(img,MetadataCatalog.get(dataset_name),scale=1)
|
58 |
out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
|
59 |
return out.get_image()
|
60 |
if file is None:
|