Update app.py

app.py

@@ -10,6 +10,7 @@ from PIL import Image
 import moviepy.editor as mp
 from infer import lotus  # Import the depth model inference function
 import logging
+import io  # Import io module
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
@@ -25,7 +26,7 @@ def preprocess_video(video_path, target_fps=24, max_resolution=(1920, 1080)):
     
     # Resize video if it's larger than the target resolution
     if video.size[0] > max_resolution[0] or video.size[1] > max_resolution[1]:
-        video = video.resize(newsize=max_resolution)
+        video = video.resize(height=max_resolution[1])
     
     # Adjust FPS if target_fps is specified
     if target_fps > 0:
@@ -41,8 +42,13 @@ def process_frame(image, seed=0):
         torch.manual_seed(seed)
         np.random.seed(seed)
         
-        # Process through the depth model (assuming lotus accepts image data)
-        _, output_d = lotus(image, 'depth', seed, device)
+        # Save image to an in-memory file
+        img_bytes = io.BytesIO()
+        image.save(img_bytes, format='PNG')
+        img_bytes.seek(0)  # Reset file pointer to the beginning
+        
+        # Process through the depth model
+        _, output_d = lotus(img_bytes, 'depth', seed, device)
         
         # Convert depth output to numpy array
         depth_array = np.array(output_d)
@@ -161,33 +167,7 @@ def process_wrapper(video, fps=0, seed=0, batch_size=16):
 
 # Custom CSS for styling
 custom_css = """
-    .title-container {
-        text-align: center;
-        padding: 10px 0;
-    }
-    
-    #title {
-        font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, Helvetica, Arial, sans-serif;
-        font-size: 36px;
-        font-weight: bold;
-        color: #000000;
-        padding: 10px;
-        border-radius: 10px;
-        display: inline-block;
-        background: linear-gradient(
-            135deg,
-            #e0f7fa, #e8f5e9, #fff9c4, #ffebee,
-            #f3e5f5, #e1f5fe, #fff3e0, #e8eaf6
-        );
-        background-size: 400% 400%;
-        animation: gradient-animation 15s ease infinite;
-    }
-    
-    @keyframes gradient-animation {
-        0% { background-position: 0% 50%; }
-        50% { background-position: 100% 50%; }
-        100% { background-position: 0% 50%; }
-    }
+    /* Your existing custom CSS */
 """
 
 # Gradio Interface