Update README.md

README.md

@@ -20,16 +20,16 @@ Okkhor Diffusion is a category of Denoising Diffusion Probabilistic Models desig
 ### Usage
 ```py
 from diffusers import DiffusionPipeline
-from typing import List, Optional, Tuple, Union
 import torch
+device="cuda"
 pipeline = DiffusionPipeline.from_pretrained(
               "ahmedfaiyaz/OkkhorDiffusion-CMATERdb",
               custom_pipeline="ahmedfaiyaz/OkkhorDiffusion",
               embedding=torch.float16
             )
-pipeline.to("cuda")
-pipeline.embedding=torch.tensor([0]) # 'প্র': 0
-pipeline(batch_size=1,num_inference_steps=1000).images[0]
+pipeline.to(device)
+pipeline.embedding=torch.tensor([0],device=device) # 'প্র': 0
+pipeline(batch_size=1,num_inference_steps=100).images[0]
 ```
 |OkkhorDiffusion-CMATERdb||
 |---------|--------|