Update app.py

app.py

@@ -11,7 +11,16 @@ repos = [
     "hsuwill000/LCM_SoteMix_OpenVINO_CPU_Space_TAESD",
     "hsuwill000/LCM_SoteMix_OpenVINO_CPU_Space_TAESD_0"
 ]
+class CustomClient(Client):
+    def __init__(self, *args, timeout=30, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.timeout = timeout
 
+    def _request(self, method, url, **kwargs):
+        # 设置 timeout 参数
+        kwargs['timeout'] = self.timeout
+        return super()._request(method, url, **kwargs)
+   
 # Counter for image filenames to avoid overwriting
 count = 0
 repo_index = 0  # This will keep track of the current repository
@@ -20,7 +29,7 @@ repo_index = 0  # This will keep track of the current repository
 def infer_gradio(prompt: str):
     global count, repo_index
     # Create a Client instance to communicate with the Hugging Face space
-    client = Client(repos[repo_index], hf_token=api_key)
+    client = CustomClient(repos[repo_index], hf_token=api_key,timeout=300)
     
     # Prepare the inputs for the prediction
     inputs = {