Update app.py

app.py

@@ -2,6 +2,7 @@ import os
 import torch
 import random
 from PIL import Image
+import cv2
 
 import gradio as gr
 from glob import glob
@@ -206,18 +207,19 @@ class AnimateController:
         generator.manual_seed(seed)
 
 
-        # Open the image
-        image = Image.open(init_image).convert("RGB")
+       
+        # Load the image
+        img = cv2.imread('image.jpg')
 
-        # Resize the image 
-        image.resize((512, 512))
+        # Resize the image to (64, 64)
+        img = cv2.resize(img, (64, 64))
 
-        # Save the resized image to the specified output path
-        image.save("resized.jpg")
+        # Save the flattened image as a JPEG file
+        cv2.imwrite('resized_image.jpg', img)
         
         sample = pipeline(
             prompt_textbox,
-            init_image          = "resized.jpg",
+            init_image          = "resized_image.jpg",
             negative_prompt     = negative_prompt_textbox,
             num_inference_steps = 25,
             guidance_scale      = 8.,