Update app.py

app.py

@@ -5,10 +5,15 @@ import time
 
 api = HfApi()
 
-api.restart_space(repo_id="HuggingFaceH4/idefics2-8b-playground")
-api.restart_space(repo_id="HuggingFaceH4/idefics2-8b-vdpoed-playground")
-while api.space_info("HuggingFaceH4/idefics2-8b-playground").runtime.stage != "RUNNING" or api.space_info("HuggingFaceH4/idefics2-8b-vdpoed-playground").runtime.stage != "RUNNING":
-    time.sleep(1)
+repo_ids = ["HuggingFaceH4/idefics2-8b-playground", "HuggingFaceH4/idefics2-8b-vdpoed-playground"]
+
+for repo_id in repo_ids:
+if api.space_info(repo_id).runtime.stage != "RUNNING":
+    api.restart_space(repo_id="HuggingFaceH4/idefics2-8b-playground")
+
+for repo_id in repo_ids:
+    if api.space_info(repo_id).runtime.stage != "RUNNING":
+        time.sleep(1)
 
 client_idefics2 = Client("HuggingFaceH4/idefics2-8b-playground")
 client_idefics2_dpoed = Client("HuggingFaceH4/idefics2-8b-vdpoed-playground")