Update README.md

README.md

@@ -3,7 +3,7 @@ license: mit
 language:
 - en
 pipeline_tag: robotics
-library_name: diffusers
+library_name: transformers
 tags:
 - robotics
 - pytorch
@@ -66,6 +66,7 @@ model = create_model(
     args=config,
     dtype=torch.bfloat16, 
     pretrained_vision_encoder_name_or_path=pretrained_vision_encoder_name_or_path,
+    pretrained='robotics-diffusion-transformer/rdt-1b',
     control_frequency=25,
 )
 # Start inference process