muhammadsalmanalfaridzi commited on
Commit
8744244
·
verified ·
1 Parent(s): dc9a186

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,7 +16,7 @@ def detect_objects(image):
16
  temp_file_path = temp_file.name
17
 
18
  # Lakukan prediksi pada gambar
19
- predictions = model.predict(temp_file_path, confidence=70, overlap=80).json()
20
 
21
  # Menghitung jumlah objek per kelas
22
  class_count = {}
@@ -38,7 +38,7 @@ def detect_objects(image):
38
  result_text += f"\nTotal Product Nestle: {total_count}"
39
 
40
  # Menyimpan gambar dengan prediksi
41
- output_image = model.predict(temp_file_path, confidence=70, overlap=80).save("/tmp/prediction.jpg")
42
 
43
  # Hapus file sementara setelah prediksi
44
  os.remove(temp_file_path)
 
16
  temp_file_path = temp_file.name
17
 
18
  # Lakukan prediksi pada gambar
19
+ predictions = model.predict(temp_file_path, confidence=60, overlap=80).json()
20
 
21
  # Menghitung jumlah objek per kelas
22
  class_count = {}
 
38
  result_text += f"\nTotal Product Nestle: {total_count}"
39
 
40
  # Menyimpan gambar dengan prediksi
41
+ output_image = model.predict(temp_file_path, confidence=60, overlap=80).save("/tmp/prediction.jpg")
42
 
43
  # Hapus file sementara setelah prediksi
44
  os.remove(temp_file_path)