Update app.py

app.py

@@ -5,9 +5,11 @@ import os
 import time
 import traceback
 import asyncio
+from huggingface_hub import HfApi
 
 # Your Hugging Face API key (ensure this is set in your environment or replace directly)
 api_key = os.getenv('MY_API_KEY')
+api = HfApi(token=api_key)
 
 # List of repos (private spaces)
 repos = [
@@ -70,6 +72,14 @@ async def infer_gradio(prompt: str):
     # Create a list of tasks (one for each repo)
     tasks = []
     for repo in repos:
+        runtime_info = api.get_space_runtime(repo)
+        if runtime_info.stage == 'SLEEPING':
+            print(f"{repo} is now SLEEPING.")
+            api.restart_space(repo_id=repo)
+        if runtime_info.stage in ['SLEEPING', 'PAUSED']:
+            print(f"{repo} is now SLEEPING or PAUSED.")
+            api.restart_space(repo_id=repo)
+            
         # Create a CustomClient instance for each repo
         client = CustomClient(repo, hf_token=api_key, timeout=300)
         task = infer_single_gradio(client, prompt)