Update app.py
Browse files
app.py
CHANGED
@@ -47,7 +47,7 @@ def visualize_prediction(img, output_dict, threshold=0.5, id2label=None):
|
|
47 |
if id2label is not None:
|
48 |
labels = [id2label[x] for x in labels]
|
49 |
|
50 |
-
plt.figure(figsize=(
|
51 |
plt.imshow(img)
|
52 |
ax = plt.gca()
|
53 |
colors = COLORS * 100
|
|
|
47 |
if id2label is not None:
|
48 |
labels = [id2label[x] for x in labels]
|
49 |
|
50 |
+
plt.figure(figsize=(25, 20))
|
51 |
plt.imshow(img)
|
52 |
ax = plt.gca()
|
53 |
colors = COLORS * 100
|