yeq6x commited on
Commit
b892cf9
·
1 Parent(s): 74197de
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -1,5 +1,5 @@
1
  import gradio as gr
2
- import spaces
3
  import torch
4
  import torch.nn.functional as F
5
  from torch.utils.data import DataLoader
@@ -68,7 +68,7 @@ def preprocess_uploaded_image(uploaded_image, image_size):
68
  return uploaded_image
69
 
70
  # ヒートマップの生成関数
71
- @spaces.GPU
72
  def get_heatmaps(source_num, x_coords, y_coords, uploaded_image):
73
  if type(uploaded_image) == str:
74
  uploaded_image = Image.open(uploaded_image)
@@ -240,10 +240,10 @@ with gr.Blocks() as demo:
240
  gr.Markdown("# Examples")
241
  gr.Examples(
242
  examples=[
243
- ["0", "50", "50", "resources/examples/2488.jpg"],
244
- ["0", "50", "50", "resources/examples/2899.jpg"]
245
  ],
246
- inputs=[source_num, x_coords, y_coords, input_image],
247
  )
248
  with gr.Column():
249
  output_plot = gr.Plot()
 
1
  import gradio as gr
2
+ # import spaces
3
  import torch
4
  import torch.nn.functional as F
5
  from torch.utils.data import DataLoader
 
68
  return uploaded_image
69
 
70
  # ヒートマップの生成関数
71
+ # @spaces.GPU
72
  def get_heatmaps(source_num, x_coords, y_coords, uploaded_image):
73
  if type(uploaded_image) == str:
74
  uploaded_image = Image.open(uploaded_image)
 
240
  gr.Markdown("# Examples")
241
  gr.Examples(
242
  examples=[
243
+ ["resources/examples/2488.jpg"],
244
+ ["resources/examples/2899.jpg"]
245
  ],
246
+ inputs=[input_image],
247
  )
248
  with gr.Column():
249
  output_plot = gr.Plot()