Update app.py

app.py

@@ -43,19 +43,19 @@ def kwg(photo):
         st.write('На фото есть дети')
     else:
         st.write('Здесь только взрослые')
-        adult = True
+        if len(ages) == 1:
+            st.write(f'И ему {ages[0]} лет')
+        else:
+            for j in range(len(ages)):
+                st.write(f'{ages[j]} лет')
+            return
 
     res = classifier(photo, candidate_labels=["kid with gun", "kid with toy", "kid with alcohol drink"])
+        
     if res[0]['label'] == "kid with gun":
         st.write('ОБОЖЕМОЙ у РЕБЕнкА ОРУЖИЕ СДЕЛАЙТЕ ЧТО-НИБУДЬ')
     elif res[0]['label'] == "kid with alcohol drink":
         st.write('ОТДАЙ ПИВО')
-    elif adult:
-        if len(ages) == 1:
-            st.write(f'И ему {ages[0]} лет')
-        else:
-            for j in range(len(ages)):
-                st.write(f'{ages[j]} лет')
     else:
         st.write('какой милый ребеночек :3')