fix: enforce dtype

app.py

@@ -23,7 +23,7 @@ def preprocess_image(image):
         transforms.ToTensor(),
         transforms.Normalize([0.5], [0.5])
     ])
-    image = preprocess(image).unsqueeze(0).to(device)
+    image = preprocess(image).unsqueeze(0).to(device, dtype=dtype)
     return image
 
 def encode_image(image, vae):
@@ -184,3 +184,4 @@ with gr.Blocks(css=css) as demo:
 demo.launch()
 
 
+