Update app.py

app.py

@@ -8,29 +8,28 @@ HF_TOKEN = os.getenv("HF_TOKEN")
 # Initialize HF API
 api = HfApi(token=HF_TOKEN)
 
-def check_zero_gpu(username: str):
+def check_spaces_zero_gpu(username: str):
     output = ""
     try:
-        repos = api.list_models(author=username, token=HF_TOKEN)
-        if not repos:
-            return f"No repositories found for user '{username}'"
+        # List all spaces for the user
+        spaces = api.list_spaces(author=username, token=HF_TOKEN)
+        if not spaces:
+            return f"No Spaces found for user '{username}'"
         
-        for repo in repos:
+        for space in spaces:
             # Default: assume ZeroGPU
             zero_gpu = "Yes"
             
-            # Check model card for hardware requirements
+            # Check if space requires GPU
             try:
-                card = api.model_info(repo_id=repo.modelId, token=HF_TOKEN)
-                # If model has 'hardware' in card, and it mentions GPU, mark as No
-                if hasattr(card, "hardware") and card.hardware:
-                    if any("gpu" in h.lower() for h in card.hardware):
+                space_info = api.space_info(space_id=space.id, token=HF_TOKEN)
+                if getattr(space_info, "hardware", None):
+                    if any("gpu" in h.lower() for h in space_info.hardware):
                         zero_gpu = "No"
             except Exception:
-                # If info not available, keep as Yes
-                pass
-
-            output += f"{repo.modelId} | ZeroGPU: {zero_gpu}\n"
+                pass  # Keep as Yes if info unavailable
+            
+            output += f"{space.id} | ZeroGPU: {zero_gpu}\n"
 
     except Exception as e:
         return f"Error: {e}"
@@ -39,11 +38,11 @@ def check_zero_gpu(username: str):
 
 # Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("## Hugging Face Repo ZeroGPU Checker")
+    gr.Markdown("## Hugging Face Spaces ZeroGPU Checker")
     username_input = gr.Textbox(label="Hugging Face Username", value="rahul7star")
-    scan_button = gr.Button("Check Repositories")
+    scan_button = gr.Button("Check Spaces")
     output_box = gr.Textbox(label="ZeroGPU Status", lines=25)
     
-    scan_button.click(check_zero_gpu, inputs=username_input, outputs=output_box)
+    scan_button.click(check_spaces_zero_gpu, inputs=username_input, outputs=output_box)
 
 demo.launch()