acmyu commited on
Commit
cbd28e2
·
1 Parent(s): 3aa6ee8

fix film bugs

Browse files
Files changed (1) hide show
  1. main.py +9 -1
main.py CHANGED
@@ -1225,11 +1225,19 @@ def run_interpolate_frames(frame1, frame2, times_to_interp):
1225
  film.setup()
1226
 
1227
  thumb_size = (512, 512)
 
1228
  frame1.thumbnail(thumb_size)
1229
  frame2.thumbnail(thumb_size)
1230
 
1231
  results = film.predict(frame1, frame2, times_to_interp)
1232
- print(results)
 
 
 
 
 
 
 
1233
  return results, getThumbnails(results)
1234
 
1235
 
 
1225
  film.setup()
1226
 
1227
  thumb_size = (512, 512)
1228
+ width, height = image.size
1229
  frame1.thumbnail(thumb_size)
1230
  frame2.thumbnail(thumb_size)
1231
 
1232
  results = film.predict(frame1, frame2, times_to_interp)
1233
+
1234
+ if not isinstance(results, list):
1235
+ results = [results]
1236
+
1237
+ results = [Image.open(str(r)) for r in results]
1238
+ for r in results:
1239
+ r.thumbnail((width, height))
1240
+
1241
  return results, getThumbnails(results)
1242
 
1243