guardiancc commited on
Commit
ee3485e
·
verified ·
1 Parent(s): 591ab7e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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((pt_src_image, pt_ref_image, pt_ref_acceleration, pt_step, pt_scale, pt_seed)
160
- return leffa_predictor.leffa_predict_pt(p))
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