Update app.py
Browse filesForgot an "s" on variable name
app.py
CHANGED
@@ -74,7 +74,7 @@ def detect_objects(model_name,url_input,image_input,threshold):
|
|
74 |
#Make prediction
|
75 |
processed_output_list = make_prediction(image, feature_extractor, model)
|
76 |
print(processed_output_list[1:])
|
77 |
-
|
78 |
|
79 |
#Visualize prediction
|
80 |
viz_img = visualize_prediction(image, processed_outputs, threshold, model.config.id2label)
|
|
|
74 |
#Make prediction
|
75 |
processed_output_list = make_prediction(image, feature_extractor, model)
|
76 |
print(processed_output_list[1:])
|
77 |
+
processed_outputs = processed_output_list[0]
|
78 |
|
79 |
#Visualize prediction
|
80 |
viz_img = visualize_prediction(image, processed_outputs, threshold, model.config.id2label)
|