marstin commited on
Commit
c59be23
·
1 Parent(s): d114c2c

[martin-dev] fix space gpu

Browse files
Files changed (1) hide show
  1. demo/launch_gradio.py +3 -0
demo/launch_gradio.py CHANGED
@@ -3,6 +3,8 @@
3
  from typing import Any, Dict, List, Optional, Tuple
4
 
5
  import gradio as gr
 
 
6
  import matplotlib.pyplot as plt
7
  import numpy as np
8
  import torch
@@ -534,6 +536,7 @@ def process_dual_inputs(
534
  return None, f'Error: {str(e)}'
535
 
536
 
 
537
  def create_demo() -> gr.Blocks:
538
  """Create and configure the Gradio demo interface for dual image comparison.
539
 
 
3
  from typing import Any, Dict, List, Optional, Tuple
4
 
5
  import gradio as gr
6
+ from spaces import GPU
7
+
8
  import matplotlib.pyplot as plt
9
  import numpy as np
10
  import torch
 
536
  return None, f'Error: {str(e)}'
537
 
538
 
539
+ @GPU
540
  def create_demo() -> gr.Blocks:
541
  """Create and configure the Gradio demo interface for dual image comparison.
542