Update app.py

app.py

@@ -1,8 +1,17 @@
 import gradio as gr
 import os
 from huggingface_hub import login, hf_hub_download
-from huggingface_hub import spaces  # <-- new import for ZeroGPU
 
+# Import spaces as a separate package (not from huggingface_hub)
+try:
+    import spaces
+    SPACES_AVAILABLE = True
+    print("Successfully imported spaces package")
+except ImportError:
+    SPACES_AVAILABLE = False
+    print("Spaces package not available, will run without GPU acceleration")
+
+# Try to authenticate explicitly
 hf_token = os.environ.get("HF_TOKEN")
 if hf_token:
     print(f"Found HF_TOKEN in environment (length: {len(hf_token)})")
@@ -14,12 +23,8 @@ if hf_token:
 else:
     print("No HF_TOKEN found in environment variables")
 
-@spaces.GPU
-def check_model_access():
-    """
-    Because we've decorated this function with @spaces.GPU,
-    a ZeroGPU session is allocated right before the function runs.
-    """
+# Define model access function
+def check_model_access_impl():
     model_id = "CohereForAI/c4ai-command-r7b-arabic-02-2025"
     try:
         print(f"Attempting to access {model_id}...")
@@ -42,26 +47,46 @@ def check_model_access():
         print(f"Error accessing model: {error_msg}")
         return f"❌ Error accessing {model_id}: {error_msg}"
 
+# Use spaces.GPU decorator if available, otherwise use function directly
+if SPACES_AVAILABLE:
+    try:
+        @spaces.GPU
+        def check_model_access():
+            return check_model_access_impl()
+    except Exception as e:
+        print(f"Error with spaces.GPU: {e}")
+        # Fallback to non-GPU version
+        def check_model_access():
+            return check_model_access_impl()
+else:
+    # No spaces package, use direct implementation
+    def check_model_access():
+        return check_model_access_impl()
+
+# Create Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# CohereForAI/c4ai-command-r7b-arabic-02-2025 Access Test")
-
+    
+    # Show status information
     auth_status = "✅ Token found" if hf_token else "❌ No token found"
-    gr.Markdown(f"**Authentication Status:** {auth_status}")
-
+    spaces_status = "✅ Available" if SPACES_AVAILABLE else "❌ Not available"
+    gr.Markdown(f"**Authentication Status:** {auth_status} | **Spaces GPU Support:** {spaces_status}")
+    
     check_btn = gr.Button("Test Model Access")
     result = gr.Textbox(label="Result", lines=10)
-
     check_btn.click(fn=check_model_access, inputs=None, outputs=result)
-
+    
     gr.Markdown("""
     ## Next Steps
+    
     If the test above is successful, it confirms you can access the model.
     To set up a fully functional demo:
-
-    1. Configure your Space to use ZeroGPU hardware in Settings → Hardware
-    2. Update your code to use the model for text generation with @spaces.GPU
-
+    
+    1. Make sure you have the `spaces` package installed in requirements.txt
+    2. Configure your Space to use ZeroGPU hardware in Settings → Hardware
+    3. Update your code to use the model for text generation with @spaces.GPU
+    
     This test simply checks if your authentication is working correctly with the model repository.
     """)
 
-demo.launch()
+demo.launch()