Spaces:

nyanko7
/

toaru-xl-model

Running on Zero

nyanko7 commited on 26 days ago

Commit

d9a5943

verified ·

1 Parent(s): efde29f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -71,7 +71,7 @@ def tpu_inference_api(
         "init_image": init_image,
         "inference_steps": inference_steps,
     }
-    response = requests.post(url, json=payload)
     if response.status_code != 200:
         raise Exception(f"Error calling API: {response.status_code} - {response.text}")

         "init_image": init_image,
         "inference_steps": inference_steps,
     }
+    response = requests.post(url, json=payload, timeout=60)
     if response.status_code != 200:
         raise Exception(f"Error calling API: {response.status_code} - {response.text}")