Update README.md

README.md

@@ -20,15 +20,15 @@ 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-Ekush",
               custom_pipeline="ahmedfaiyaz/OkkhorDiffusion",
-              embedding=torch.float16
+              embedding=torch.int16
             )
-pipeline.to("cuda")
-pipeline.embedding=torch.tensor([10]) # 'ও': 10
+pipeline.to(device)
+pipeline.embedding=torch.tensor([10],device=device) # 'ও': 10
 pipeline(batch_size=1,num_inference_steps=1000).images[0]
 ```