sjtu-deepvision commited on
Commit
7d1a82d
·
verified ·
1 Parent(s): 7cdacae

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,10 +1,10 @@
 
1
  import os
2
  import numpy as np
3
  import torch
4
  from PIL import Image
5
  import gradio as gr
6
  from gradio_imageslider import ImageSlider
7
- import spaces # 必须放在最前面,确保 ZeroGPU 初始化
8
 
9
  # 延迟 CUDA 初始化
10
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -90,7 +90,7 @@ def create_gradio_interface():
90
  inputs=input_image,
91
  outputs=output_slider,
92
  fn=process_image,
93
- cache_examples=False, # 缓存结果以加快加载速度
94
  label="Example Images",
95
  )
96
 
 
1
+ import spaces # 必须放在最前面
2
  import os
3
  import numpy as np
4
  import torch
5
  from PIL import Image
6
  import gradio as gr
7
  from gradio_imageslider import ImageSlider
 
8
 
9
  # 延迟 CUDA 初始化
10
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
90
  inputs=input_image,
91
  outputs=output_slider,
92
  fn=process_image,
93
+ cache_examples=True, # 缓存结果以加快加载速度
94
  label="Example Images",
95
  )
96