update device

app.py

@@ -31,7 +31,16 @@ def load_model():
             unet=None,
             token=token,
         )
-        print("Model loaded successfully!")
+        
+        # Set the execution device to match the text encoder's device
+        # This ensures tokenizer outputs go to the correct device
+        if hasattr(text_encoder, 'device'):
+            text_pipe._execution_device = text_encoder.device
+        else:
+            # Find the device of the first parameter
+            text_pipe._execution_device = next(text_encoder.parameters()).device
+        
+        print(f"Model loaded successfully on device: {text_pipe._execution_device}")
     return text_pipe
 
 @spaces.GPU