Update app.py

app.py

@@ -27,19 +27,8 @@ model = snapshot_download(repo_id="akhaliq/frame-interpolation-film-style")
 interpolator = interpolator.Interpolator(model, None)
 
 
-def generate_story(choice, input_text):
-    query = "<BOS> <{0}> {1}".format(choice, input_text)
-    
-    print(query)
-    generated_text = story_gen(query)
-    generated_text = generated_text[0]['generated_text']
-    generated_text = generated_text.split('> ')[2]
-    
-    return generated_text
-
-
-def generate_images(text, width=128, height=128, steps=40, num_images=1,
-                    diversity=2):
+def generate_images(text, width=256, height=256, steps=50, num_images=2,
+                    diversity=4):
 
     image_bytes = image_gen(text, steps, width, height, num_images, diversity)
     
@@ -56,23 +45,24 @@ def generate_images(text, width=128, height=128, steps=40, num_images=1,
 
 
 def generate_interpolation(text, fps=5):    
-    generated_images = []
-    for t in text.split(','):
-      generated_images.extend(generate_images(t.strip()))
-    
-    frames = []
-    for i, g in enumerate(generated_images):
-      frames.append(f'frame_{i}.png')
-      g.save(frames[-1])
-    
-    vid = list(util.interpolate_recursively_from_files(frames, np.max([4, len(generated_images)]), interpolator))
-    mediapy.write_video("out.mp4", vid, fps=fps)
-    return "out.mp4"
+  images = []
+  frames = []
+  for i, t in enumerate(text.split(', ')):
+    print(f'image {i}: {t.lower().strip()}', end='...')
+    images.extend(generate_images(t.lower().strip()))
+    print('done!')
+  
+    frames.append(f'frame_{i}.png')
+    images[-1].save(frames[-1])
+
+  vid = list(util.interpolate_recursively_from_files(frames, np.max([4, len(images), fps]), interpolator))
+  mediapy.write_video("out.mp4", vid, fps=fps)
+  return "out.mp4"
 
 demo = gr.Blocks()
 
 with demo:
-  text = gr.Textbox(placeholder='human, human head, brain, brain in a computer, humanoid robot', label='Input a comma-separated list of terms:')
+  text = gr.Textbox(placeholder='human, brain in a computer, humanoid robot', label='Input a comma-separated list of terms:')
   button = gr.Button("Generate Video")
   output = gr.Video(label="Generated Video")