Spaces:
Running
Running
IdleCloud commited on
Commit ·
8d98638
1
Parent(s): 81dec11
Fix custom job polling route
Browse files- app.py +2 -0
- tests/test_image_inversion_job_api.py +1 -0
app.py
CHANGED
|
@@ -950,5 +950,7 @@ if __name__ == "__main__":
|
|
| 950 |
demo.queue(max_size=8).launch(
|
| 951 |
server_name="0.0.0.0",
|
| 952 |
server_port=7860,
|
|
|
|
|
|
|
| 953 |
app_kwargs={"lifespan": JOB_API_LIFESPAN},
|
| 954 |
)
|
|
|
|
| 950 |
demo.queue(max_size=8).launch(
|
| 951 |
server_name="0.0.0.0",
|
| 952 |
server_port=7860,
|
| 953 |
+
# 关闭 Gradio SSR,避免其 SvelteKit 中间件在 FastAPI 路由匹配前截获自定义 GET API。
|
| 954 |
+
ssr_mode=False,
|
| 955 |
app_kwargs={"lifespan": JOB_API_LIFESPAN},
|
| 956 |
)
|
tests/test_image_inversion_job_api.py
CHANGED
|
@@ -762,6 +762,7 @@ class ImageInversionJobAPITest(unittest.TestCase):
|
|
| 762 |
if isinstance(key, ast.Constant) and isinstance(key.value, str)
|
| 763 |
}
|
| 764 |
self.assertEqual(result_keys, set(EXPECTED_RESULT))
|
|
|
|
| 765 |
self.assertIn('app_kwargs={"lifespan": JOB_API_LIFESPAN}', app_source)
|
| 766 |
self.assertIn('token=False', app_source)
|
| 767 |
self.assertIn('HF_HUB_DISABLE_IMPLICIT_TOKEN', app_source)
|
|
|
|
| 762 |
if isinstance(key, ast.Constant) and isinstance(key.value, str)
|
| 763 |
}
|
| 764 |
self.assertEqual(result_keys, set(EXPECTED_RESULT))
|
| 765 |
+
self.assertIn("ssr_mode=False", app_source)
|
| 766 |
self.assertIn('app_kwargs={"lifespan": JOB_API_LIFESPAN}', app_source)
|
| 767 |
self.assertIn('token=False', app_source)
|
| 768 |
self.assertIn('HF_HUB_DISABLE_IMPLICIT_TOKEN', app_source)
|