Update app.py

app.py

@@ -42,9 +42,16 @@ class AudioSeparatorD:
             "cuda_version": torch.version.cuda if torch.cuda.is_available() else "N/A",
             "mps_available": hasattr(torch.backends, "mps") and torch.backends.mps.is_available(),
             "device": "cuda" if torch.cuda.is_available() else ("mps" if hasattr(torch.backends, "mps") and torch.backends.mps.is_available() else "cpu"),
-            "memory_total": torch.cuda.get_device_properties(0).total_memory if torch.cuda.is_available() else 0,
-            "memory_allocated": torch.cuda.memory_allocated() if torch.cuda.is_available() else 0
         }
+        
+        # Only add memory info if CUDA is available
+        if torch.cuda.is_available():
+            info["memory_total"] = torch.cuda.get_device_properties(0).total_memory
+            info["memory_allocated"] = torch.cuda.memory_allocated()
+        else:
+            info["memory_total"] = 0
+            info["memory_allocated"] = 0
+            
         return info
     
     def analyze_audio_characteristics(self, audio_file: str) -> Dict:
@@ -468,7 +475,11 @@ class AudioSeparatorD:
                 
                 # Set default model if not specified
                 if model_name is None:
-                    model_name = "model_bs_roformer_ep_317_sdr_12.9755.ckpt"
+                    models = self.get_available_models()
+                    if models:
+                        model_name = list(models.keys())[0]  # Use first available model
+                    else:
+                        return False, "No models available"
                 
                 # Initialize separator with updated parameters
                 self.separator = Separator(
@@ -601,7 +612,7 @@ class AudioSeparatorD:
         
         return custom_params
     
-    def (self):
+    def get_phistory(self):
         """Get enhanced processing history with analytics"""
         if not self.processing_history:
             return "No processing history available"
@@ -688,7 +699,7 @@ def create_interface():
                 # Model dropdown with enhanced display
                 model_dropdown = gr.Dropdown(
                     choices=list(model_list.keys()) if model_list else [],
-                    value="model_bs_roformer_ep_317_sdr_12.9755.ckpt" if model_list else None,
+                    value=list(model_list.keys())[0] if model_list else None,
                     label="🤖 AI Model Selection",
                     info="Choose an AI model or use auto-selection",
                     elem_id="model_dropdown"
@@ -1186,7 +1197,7 @@ if __name__ == "__main__":
     interface = create_interface()
     interface.launch(
         server_port=7860,
-        theme=" NeoPy/Soft",
+        theme="NeoPy/Soft", 
         share=True,
         debug=True
     )