IdleCloud commited on
Commit
8d98638
·
1 Parent(s): 81dec11

Fix custom job polling route

Browse files
Files changed (2) hide show
  1. app.py +2 -0
  2. 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)