Update app.py

app.py

@@ -76,7 +76,7 @@ def run_comparison(iterations: int, seed: int, use_deterministic: bool, device:
     """
     
     # Set device environment variable for subprocess
-    # On Hugging Face Spaces, check GPU availability more carefully
+    # On Hugging Face Spaces with GPU, try to use CUDA
     if device == "cuda":
         try:
             import torch
@@ -85,18 +85,21 @@ def run_comparison(iterations: int, seed: int, use_deterministic: bool, device:
                 try:
                     # Try to get device name to verify GPU works
                     gpu_name = torch.cuda.get_device_name(0)
-                    print(f"✅ GPU available: {gpu_name}")
+                    gpu_count = torch.cuda.device_count()
+                    print(f"✅ GPU available: {gpu_name} (Count: {gpu_count})")
+                    # Keep device as "cuda"
                 except Exception as e:
-                    print(f"⚠️ GPU detection failed: {e}, falling back to CPU")
-                    device = "cpu"
+                    print(f"⚠️ GPU detection failed: {e}")
+                    print("   Attempting to use CUDA anyway (may work)...")
+                    # Don't fallback immediately - let it try
             else:
-                print("⚠️ CUDA not available, using CPU")
+                print("⚠️ CUDA not available, falling back to CPU")
                 device = "cpu"
         except ImportError:
-            print("⚠️ PyTorch not available, using CPU")
+            print("⚠️ PyTorch not available, falling back to CPU")
             device = "cpu"
         except Exception as e:
-            print(f"⚠️ GPU check error: {e}, using CPU")
+            print(f"⚠️ GPU check error: {e}, falling back to CPU")
             device = "cpu"
     
     # Set environment variable for subprocess to pick up
@@ -168,25 +171,33 @@ def run_comparison(iterations: int, seed: int, use_deterministic: bool, device:
 
 
 def check_gpu():
-    """Check if GPU is available."""
+    """Check if GPU is available on Hugging Face Spaces."""
     try:
         import torch
+        
+        # Check CUDA availability
         if torch.cuda.is_available():
             try:
                 gpu_name = torch.cuda.get_device_name(0)
                 gpu_count = torch.cuda.device_count()
-                return f"✅ GPU Available: {gpu_name} (Count: {gpu_count})"
+                cuda_version = torch.version.cuda
+                return f"✅ GPU Available: {gpu_name} (Count: {gpu_count}, CUDA: {cuda_version})"
             except Exception as e:
-                return f"⚠️ GPU detected but error accessing: {str(e)}"
+                # GPU might be available but not immediately accessible
+                return f"✅ GPU Detected (accessing: {str(e)[:50]}...)"
         else:
-            # Check if we're on Hugging Face Spaces
+            # On Hugging Face Spaces, check environment
             if os.getenv("SPACE_ID"):
-                return "⚠️ No GPU available on this Space. Please upgrade to GPU tier."
-            return "⚠️ No GPU available, using CPU"
+                # Check if GPU hardware is allocated
+                hf_hardware = os.getenv("SPACE_HARDWARE", "unknown")
+                if "gpu" in hf_hardware.lower() or "t4" in hf_hardware.lower() or "l4" in hf_hardware.lower():
+                    return f"⚠️ GPU Hardware ({hf_hardware}) allocated but not accessible yet. Try running anyway."
+                return f"⚠️ No GPU on this Space (hardware: {hf_hardware}). Please configure GPU tier."
+            return "⚠️ No GPU available, will use CPU"
     except ImportError:
         return "⚠️ PyTorch not installed"
     except Exception as e:
-        return f"⚠️ Could not check GPU status: {str(e)}"
+        return f"⚠️ GPU check error: {str(e)}"
 
 
 # Create Gradio interface
@@ -240,10 +251,10 @@ with gr.Blocks(title="MentorFlow - Strategy Comparison") as demo:
             )
             
             device = gr.Radio(
-                choices=["cpu", "cuda"],
-                value="cpu",  # Default to CPU for reliability on HF Spaces
+                choices=["cuda", "cpu"],
+                value="cuda",  # Default to GPU for HF Spaces with Nvidia 4xL4
                 label="Device",
-                info="CPU (recommended) or CUDA/GPU if available on your Space"
+                info="GPU (cuda) recommended for Nvidia 4xL4, CPU fallback available"
             )
         
         with gr.Column():