guardiancc commited on
Commit
0bb596a
·
verified ·
1 Parent(s): ee3485e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -154,7 +154,7 @@ class LeffaPredictor(object):
154
  return self.leffa_predict(src_image_path, ref_image_path, "pose_transfer", ref_acceleration, step, scale, seed)
155
 
156
 
157
- @spaces.GPU(duration=120)
158
  def runPrediction(pt_src_image, pt_ref_image, pt_ref_acceleration, pt_step, pt_scale, pt_seed):
159
  leffa_predictor = LeffaPredictor()
160
  return leffa_predictor.leffa_predict_pt(pt_src_image, pt_ref_image, pt_ref_acceleration, pt_step, pt_scale, pt_seed)
 
154
  return self.leffa_predict(src_image_path, ref_image_path, "pose_transfer", ref_acceleration, step, scale, seed)
155
 
156
 
157
+ @spaces.GPU
158
  def runPrediction(pt_src_image, pt_ref_image, pt_ref_acceleration, pt_step, pt_scale, pt_seed):
159
  leffa_predictor = LeffaPredictor()
160
  return leffa_predictor.leffa_predict_pt(pt_src_image, pt_ref_image, pt_ref_acceleration, pt_step, pt_scale, pt_seed)