fix error bug

app.py

@@ -110,6 +110,9 @@ def load_data(file_path) -> np.ndarray:
 
 def display_crop(data, x_offset, y_offset, auto_contrast) -> Image:
 
+    if data is None:
+        return None
+        
     crop = data[y_offset:y_offset + 1024, x_offset:x_offset + 1024]
     original_image_normalized = Auto_contrast(crop) if auto_contrast else (crop - crop.min()) / (crop.max() - crop.min())
     input_image = Image.fromarray((original_image_normalized * 255).astype(np.uint8))
@@ -119,6 +122,9 @@ def display_crop(data, x_offset, y_offset, auto_contrast) -> Image:
 @spaces.GPU
 def process_and_denoise(data, x_offset, y_offset, auto_contrast) -> Image:
 
+    if data is None:
+        return None
+        
     crop = data[y_offset:y_offset + 1024, x_offset:x_offset + 1024]
     denoised_data = denoiser.inference(Image.fromarray(crop))