Update app.py

app.py

@@ -48,9 +48,9 @@ def generate_and_export_animation(prompt):
         output = pipe(
             prompt=prompt,
             negative_prompt="bad quality, worse quality, low resolution, blur",
-            num_frames=16,
+            num_frames=32,
             guidance_scale=2.0,
-            num_inference_steps=6
+            num_inference_steps=18
         )
         print('Video frames generated')
         
@@ -58,18 +58,7 @@ def generate_and_export_animation(prompt):
         with tempfile.NamedTemporaryFile(suffix='.mp4', delete=False) as temp_file:
             temp_video_path = temp_file.name
             print('temp_video_path', temp_video_path)
-
-            processed_frames = []
-            for frame_tensor in output.frames:
-                # Convert frame tensor to numpy array
-                frame_numpy = frame_tensor.numpy()
-            
-                # Append processed frame to the list
-                processed_frames.append(frame_numpy)
-
-            # Export processed frames to a video file
-            export_to_video(processed_frames, temp_video_path)
-            # export_to_video(output.frames, temp_video_path)
+            export_to_video(output.frames[0], temp_video_path)
 
         with open(temp_video_path, 'rb') as video_file:
             video_binary = video_file.read()