ZhengChong commited on
Commit
8d0429d
1 Parent(s): 6a6227f

Add spaces library for GPU support

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. requirements.txt +1 -0
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import argparse
2
  import os
3
  from datetime import datetime
4
-
5
  import gradio as gr
6
  import numpy as np
7
  import torch
@@ -126,7 +126,7 @@ automasker = AutoMasker(
126
  device='cuda',
127
  )
128
 
129
-
130
  def submit_function(
131
  person_image,
132
  cloth_image,
 
1
  import argparse
2
  import os
3
  from datetime import datetime
4
+ import spaces
5
  import gradio as gr
6
  import numpy as np
7
  import torch
 
126
  device='cuda',
127
  )
128
 
129
+ @spaces.GPU
130
  def submit_function(
131
  person_image,
132
  cloth_image,
requirements.txt CHANGED
@@ -1,6 +1,7 @@
1
  accelerate==0.31.0
2
  diffusers==0.29.2
3
  gradio==4.39.0
 
4
  huggingface_hub==0.23.4
5
  matplotlib==3.9.1
6
  numpy==1.26.4
 
1
  accelerate==0.31.0
2
  diffusers==0.29.2
3
  gradio==4.39.0
4
+ spaces==0.29.2
5
  huggingface_hub==0.23.4
6
  matplotlib==3.9.1
7
  numpy==1.26.4