Update app.py

app.py

@@ -6,7 +6,7 @@ import cv2
 import matplotlib.pyplot as plt
 
 # Konfigurasi model
-MODEL_PATH = "model.pt"  # Pastikan model ada di direktori ini
+MODEL_PATH = "best.pt"  # Pastikan model ada di direktori ini
 CLASS_NAMES = ["bag", "person-static-object"]  # Sesuaikan dengan kelas Anda
 
 # Fungsi untuk memuat model
@@ -46,7 +46,7 @@ uploaded_file = st.file_uploader("Pilih gambar...", type=["jpg", "png", "jpeg"])
 if uploaded_file is not None:
     # Memuat gambar
     image = Image.open(uploaded_file).convert("RGB")
-    st.image(image, caption="Gambar Input", use_column_width=True)
+    st.image(image, caption="Gambar Input", use_container_width=True)
     
     # Proses deteksi
     if st.button("Deteksi Objek"):