superuser-aisensum commited on
Commit
c0871ec
·
verified ·
1 Parent(s): 156a0ed

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 = {}
@@ -33,7 +33,7 @@ def detect_objects(image):
33
  result_text += f"{class_name}: {count} objek\n"
34
 
35
  # Menyimpan gambar dengan prediksi
36
- output_image = model.predict(temp_file_path, confidence=70, overlap=80).save("/tmp/prediction.jpg")
37
 
38
  # Hapus file sementara setelah prediksi
39
  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 = {}
 
33
  result_text += f"{class_name}: {count} objek\n"
34
 
35
  # Menyimpan gambar dengan prediksi
36
+ output_image = model.predict(temp_file_path, confidence=60, overlap=80).save("/tmp/prediction.jpg")
37
 
38
  # Hapus file sementara setelah prediksi
39
  os.remove(temp_file_path)