kevinconka commited on
Commit
69346b7
·
1 Parent(s): 5f5664d

remove zero gpu

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ h1 {
46
  }
47
  """
48
 
49
- model = load_model("ahoy-RGB-b2", device="cuda")
50
 
51
  @spaces.GPU
52
  def inference(image):
 
46
  }
47
  """
48
 
49
+ model = load_model("ahoy-RGB-b2")
50
 
51
  @spaces.GPU
52
  def inference(image):