Update app.py
Browse files
app.py
CHANGED
@@ -43,7 +43,7 @@ def visualize_prediction(pil_img, output_dict, threshold=0.5, id2label=None):
|
|
43 |
if id2label is not None:
|
44 |
labels = [id2label[x] for x in labels]
|
45 |
|
46 |
-
plt.figure(figsize=(
|
47 |
plt.imshow(pil_img)
|
48 |
ax = plt.gca()
|
49 |
colors = COLORS * 100
|
|
|
43 |
if id2label is not None:
|
44 |
labels = [id2label[x] for x in labels]
|
45 |
|
46 |
+
plt.figure(figsize=(18, 13))
|
47 |
plt.imshow(pil_img)
|
48 |
ax = plt.gca()
|
49 |
colors = COLORS * 100
|