Update app.py

app.py

@@ -79,7 +79,7 @@ async def infer_gradio(prompt: str):
         if runtime_info.stage in ['SLEEPING', 'PAUSED']:
             print(f"{repo} is now SLEEPING or PAUSED.")
             api.restart_space(repo_id=repo)
-            
+            break    
         # Create a CustomClient instance for each repo
         client = CustomClient(repo, hf_token=api_key, timeout=300)
         task = infer_single_gradio(client, prompt)