Z User commited on
Commit
cfa73db
·
1 Parent(s): b634edb

Fix: ValueError API endpoints must have type hints

Browse files

- Wrapped gr.api() calls in try-except for Gradio version compatibility
- Newer Gradio versions have stricter type hint requirements for class methods
- App will now start even if API registration fails (non-critical feature)

Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -2688,9 +2688,12 @@ with gr.Blocks(elem_id="main", fill_width=True, fill_height=False) as app:
2688
 
2689
  yield from _generate_image(argv)
2690
 
2691
- # Register APIs
2692
- gr.api(generate_image, api_name="generate_image", api_visibility="public", queue=True, concurrency_id="gpu")
2693
- gr.api(generate_image_stream, api_name="generate_image_stream", api_visibility="public", queue=True, concurrency_id="gpu")
 
 
 
2694
 
2695
  gr.DuplicateButton(value="Duplicate Space for private use (This demo does not work on CPU. Requires GPU Space)")
2696
 
 
2688
 
2689
  yield from _generate_image(argv)
2690
 
2691
+ # Register APIs (wrapped in try-except for Gradio version compatibility)
2692
+ try:
2693
+ gr.api(generate_image, api_name="generate_image", api_visibility="public", queue=True, concurrency_id="gpu")
2694
+ gr.api(generate_image_stream, api_name="generate_image_stream", api_visibility="public", queue=True, concurrency_id="gpu")
2695
+ except (ValueError, TypeError) as e:
2696
+ print(f"[Warning] API endpoint registration skipped: {e}")
2697
 
2698
  gr.DuplicateButton(value="Duplicate Space for private use (This demo does not work on CPU. Requires GPU Space)")
2699