Update app.py
Browse filesTrying to get lists to print
app.py
CHANGED
@@ -42,6 +42,8 @@ def visualize_prediction(pil_img, output_dict, threshold=0.7, id2label=None):
|
|
42 |
if id2label is not None:
|
43 |
labels = [id2label[x] for x in labels]
|
44 |
|
|
|
|
|
45 |
plt.figure(figsize=(16, 10))
|
46 |
plt.imshow(pil_img)
|
47 |
ax = plt.gca()
|
@@ -73,7 +75,7 @@ def detect_objects(model_name,url_input,image_input,threshold):
|
|
73 |
|
74 |
#Make prediction
|
75 |
processed_output_list = make_prediction(image, feature_extractor, model)
|
76 |
-
print(processed_output_list
|
77 |
processed_outputs = processed_output_list[0]
|
78 |
|
79 |
#Visualize prediction
|
|
|
42 |
if id2label is not None:
|
43 |
labels = [id2label[x] for x in labels]
|
44 |
|
45 |
+
print("Labels " + str(labels))
|
46 |
+
|
47 |
plt.figure(figsize=(16, 10))
|
48 |
plt.imshow(pil_img)
|
49 |
ax = plt.gca()
|
|
|
75 |
|
76 |
#Make prediction
|
77 |
processed_output_list = make_prediction(image, feature_extractor, model)
|
78 |
+
print("After make_prediction" + str(processed_output_list))
|
79 |
processed_outputs = processed_output_list[0]
|
80 |
|
81 |
#Visualize prediction
|