Krokodilpirat commited on
Commit
7b17cbd
·
verified ·
1 Parent(s): 95f2846

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +24 -2
app.py CHANGED
@@ -4,6 +4,28 @@ os.environ["HF_HOME"] = "/tmp/huggingface"
4
  os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface/transformers"
5
  os.environ["MPLCONFIGDIR"] = "/tmp/matplotlib"
6
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
7
  import gc
8
  import torch
9
  import cv2
@@ -156,7 +178,7 @@ def construct_demo():
156
  input_video = gr.Video(label="Input Video")
157
  with gr.Column(scale=2):
158
  with gr.Row(equal_height=True):
159
- # Removed loop parameter which is not supported in Gradio 4.36.0
160
  processed_video = gr.Video(label="Preprocessed Video", interactive=False, autoplay=True, show_share_button=True, scale=5)
161
  depth_vis_video = gr.Video(label="Generated Depth Video", interactive=False, autoplay=True, show_share_button=True, scale=5)
162
  stitched_video = gr.Video(label="Stitched RGBD Video", interactive=False, autoplay=True, show_share_button=True, scale=5)
@@ -175,7 +197,7 @@ def construct_demo():
175
  with gr.Column(scale=2):
176
  pass
177
 
178
- # Examples-Block wurde entfernt
179
 
180
  generate_btn.click(
181
  fn=infer_video_depth,
 
4
  os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface/transformers"
5
  os.environ["MPLCONFIGDIR"] = "/tmp/matplotlib"
6
 
7
+ # Patching the schema handling problem in Gradio 5.x
8
+ # This needs to be done before any Gradio imports
9
+ import sys
10
+ def patch_gradio_utils():
11
+ try:
12
+ from gradio_client import utils
13
+ original_get_type = utils.get_type
14
+
15
+ def patched_get_type(schema):
16
+ if isinstance(schema, bool):
17
+ return "boolean"
18
+ if not isinstance(schema, dict):
19
+ return "any"
20
+ return original_get_type(schema)
21
+
22
+ utils.get_type = patched_get_type
23
+ print("Successfully patched Gradio utils.get_type")
24
+ except Exception as e:
25
+ print(f"Could not patch Gradio utils: {e}")
26
+
27
+ patch_gradio_utils()
28
+
29
  import gc
30
  import torch
31
  import cv2
 
178
  input_video = gr.Video(label="Input Video")
179
  with gr.Column(scale=2):
180
  with gr.Row(equal_height=True):
181
+ # Removed unsupported parameters for Gradio 5.x
182
  processed_video = gr.Video(label="Preprocessed Video", interactive=False, autoplay=True, show_share_button=True, scale=5)
183
  depth_vis_video = gr.Video(label="Generated Depth Video", interactive=False, autoplay=True, show_share_button=True, scale=5)
184
  stitched_video = gr.Video(label="Stitched RGBD Video", interactive=False, autoplay=True, show_share_button=True, scale=5)
 
197
  with gr.Column(scale=2):
198
  pass
199
 
200
+ # Removed Examples block to improve loading time
201
 
202
  generate_btn.click(
203
  fn=infer_video_depth,