Adding the string because the image is working
Browse files
app.py
CHANGED
@@ -73,7 +73,19 @@ def detect_objects(model_name,url_input,image_input,threshold):
|
|
73 |
|
74 |
render = render_result(model=model, image=image_input, result=results[0])
|
75 |
|
76 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
77 |
else:
|
78 |
|
79 |
#Extract model and feature extractor
|
|
|
73 |
|
74 |
render = render_result(model=model, image=image_input, result=results[0])
|
75 |
|
76 |
+
final_str = ""
|
77 |
+
final_str_abv = ""
|
78 |
+
final_str_else = ""
|
79 |
+
|
80 |
+
for r in results:
|
81 |
+
if r.boxes.conf >= threshold:
|
82 |
+
final_str_abv =+ str(r.boxes) + "\n"
|
83 |
+
else:
|
84 |
+
final_str_else =+ str(r.boxes) + "\n"
|
85 |
+
|
86 |
+
final_str = "{:*^50}\n".format("ABOVE THRESHOLD OR EQUAL") + final_str_abv + "\n{:*^50}\n".format("BELOW THRESHOLD")+final_str_else
|
87 |
+
|
88 |
+
return render, final_str
|
89 |
else:
|
90 |
|
91 |
#Extract model and feature extractor
|