Spaces:
Paused
Paused
fix film bugs
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
|