Spaces:
Build error
Build error
Update app.py
Browse files
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=
|
| 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=
|
| 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)
|