Update app.py

app.py

@@ -6,7 +6,7 @@ import os
 # Inisialisasi Roboflow
 rf = Roboflow(api_key="Otg64Ra6wNOgDyjuhMYU")
 project = rf.workspace("alat-pelindung-diri").project("nescafe-4base")
-model = project.version(16).model
+model = project.version(36).model
 
 # Fungsi untuk menangani input dan output gambar
 def detect_objects(image):
@@ -16,7 +16,7 @@ def detect_objects(image):
         temp_file_path = temp_file.name
 
     # Lakukan prediksi pada gambar
-    predictions = model.predict(temp_file_path, confidence=50, overlap=80).json()
+    predictions = model.predict(temp_file_path, confidence=70, overlap=80).json()
     
     # Menghitung jumlah objek per kelas
     class_count = {}
@@ -33,7 +33,7 @@ def detect_objects(image):
         result_text += f"{class_name}: {count} objek\n"
     
     # Menyimpan gambar dengan prediksi
-    output_image = model.predict(temp_file_path, confidence=50, overlap=80).save("/tmp/prediction.jpg")
+    output_image = model.predict(temp_file_path, confidence=70, overlap=80).save("/tmp/prediction.jpg")
     
     # Hapus file sementara setelah prediksi
     os.remove(temp_file_path)