Update app.py

app.py

@@ -1,6 +1,14 @@
 import gradio as gr
 from gradio_client import Client, handle_file
+from huggingface_hub import HfApi
+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)
 
 client_idefics2 = Client("HuggingFaceH4/idefics2-8b-playground")
 client_idefics2_dpoed = Client("HuggingFaceH4/idefics2-8b-vdpoed-playground")