Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -156,8 +156,8 @@ class LeffaPredictor(object):
|
|
| 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(
|
| 161 |
|
| 162 |
if __name__ == "__main__":
|
| 163 |
|
|
|
|
| 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)
|
| 161 |
|
| 162 |
if __name__ == "__main__":
|
| 163 |
|