fffiloni commited on
Commit
07dc410
·
1 Parent(s): 98c6cb8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -210,12 +210,12 @@ def infer():
210
  transform = T.ToPILImage()
211
 
212
  # convert the tensor to PIL image using above transform
213
- img = transform(frame[100])
214
 
215
  # display the PIL image
216
  #img.show()
217
  img.save('frame_input.jpg')
218
- res = get_warp_res(frame[100], predicted_flow, fname_output='warped.png')
219
  #print(res)
220
  return "done", "predicted_flow.jpg", ["flofile.flo"], 'frame_input.jpg'
221
  ####################################
 
210
  transform = T.ToPILImage()
211
 
212
  # convert the tensor to PIL image using above transform
213
+ img = transform(frames[100])
214
 
215
  # display the PIL image
216
  #img.show()
217
  img.save('frame_input.jpg')
218
+ res = get_warp_res(frames[100], predicted_flow, fname_output='warped.png')
219
  #print(res)
220
  return "done", "predicted_flow.jpg", ["flofile.flo"], 'frame_input.jpg'
221
  ####################################