Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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
|
| 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)
|