Update app.py

app.py

@@ -52,7 +52,12 @@ def model_select(selected_file):
     del state_dict
     return
 
-
+def decode_data_uri_to_image(data_uri):
+    # parse the data uri
+    header, encoded = data_uri.split(",", 1)
+    data = base64.b64decode(encoded)
+    img = Image.open(BytesIO(data))
+    return img
 
 # ----------------------------- FRAME INTERPOLATION ---------------------------------
 # we cannot afford to use AI-based algorithms such as FILM or ST-MFNet,
@@ -190,7 +195,7 @@ max_64_bit_int = 2**63 - 1
 
 def sample(
     secret_token: str,
-    image: Image,
+    input_image_base64: Image,
     seed: Optional[int] = 42,
     randomize_seed: bool = False,
     motion_bucket_id: int = 80,
@@ -207,6 +212,8 @@ def sample(
         raise gr.Error(
             f'Invalid secret token. Please fork the original space if you want to use it for yourself.')
 
+    image = decode_data_uri_to_image(input_image_base64)
+    
     if image.mode == "RGBA":
         image = image.convert("RGB")