Update app.py

app.py

@@ -203,11 +203,28 @@ hubert_model = load_hubert()
 logging.info("Memuat model RMVPE...")
 rmvpe_model = RMVPE("rmvpe.pt", config.is_half, config.device)
 
+import zipfile  # Tambahkan impor ini di bagian atas file
+
 def download_model(url: str, model_name: str):
-    """Mengunduh dan menyimpan model ke direktori weights."""
-    output_path = MODEL_ROOT / model_name
-    output_path.mkdir(exist_ok=True)
-    return download_file(url, output_path / Path(url).name)
+    """Mengunduh, menyimpan, dan mengekstrak model ke direktori weights."""
+    try:
+        output_path = MODEL_ROOT / model_name
+        output_path.mkdir(exist_ok=True)
+        downloaded_file = download_file(url, output_path / "ekstrak.zip")
+        
+        # Ekstrak file ZIP ke direktori weights/{nama_model}
+        with zipfile.ZipFile(downloaded_file, 'r') as zip_ref:
+            zip_ref.extractall(output_path)
+            logging.info(f"File ZIP diekstrak ke: {output_path}")
+        
+        # Hapus file ZIP setelah ekstraksi (opsional)
+        os.remove(downloaded_file)
+        logging.info(f"File ZIP {downloaded_file} dihapus setelah ekstraksi")
+        
+        return f"Model {model_name} berhasil diunduh dan diekstrak ke {output_path}"
+    except Exception as e:
+        logging.error(f"Error saat mengunduh atau mengekstrak model: {e}")
+        raise
 
 # Antarmuka Gradio
 initial_md = """