Update app.py

app.py

@@ -20,17 +20,22 @@ def detect_objects(image):
     
     # Menghitung jumlah objek per kelas
     class_count = {}
+    total_count = 0  # Menyimpan total jumlah objek
+
     for prediction in predictions['predictions']:
         class_name = prediction['class']
         if class_name in class_count:
             class_count[class_name] += 1
         else:
             class_count[class_name] = 1
+        total_count += 1  # Tambah jumlah objek untuk setiap prediksi
 
     # Menyusun output berupa string hasil perhitungan
-    result_text = "Jumlah objek per kelas:\n"
+    result_text = "Product Nestle:\n"
     for class_name, count in class_count.items():
-        result_text += f"{class_name}: {count} objek\n"
+        result_text += f"{class_name}: {count} \n"
+    
+    result_text += f"\nTotal Product Nestle: {total_count}" 
     
     # Menyimpan gambar dengan prediksi
     output_image = model.predict(temp_file_path, confidence=60, overlap=80).save("/tmp/prediction.jpg")