Update README.md

README.md

@@ -2,13 +2,13 @@
 tags:
 - text-to-image
 - stable-diffusion
+- kviai
+- midjourney
 - lora
 - dalle-3
 - dalle
 - deepvision
 - diffusers
-- template:sd-lora
-- openskyml
 widget:
 - text: reimagine the ZX Spectrum Game MANIC MINER as a 3D modern style game
   output:
@@ -40,10 +40,56 @@ library_name: diffusers
 
 ## Model description 
 
-This is a test model very similar to Dall•E 3.
+This is a test model like Dall-E 3.
+
+Estimated generateion time is ~ 40 seconds on gpu
 
 By KVI Kontent
 
+## Usage
+
+You can try out model using Huggingface Interface API, here:
+```Python
+import requests
+import io
+from PIL import *
+
+API_URL = "https://api-inference.huggingface.co/models/Kvikontent/kviimager2.0"
+headers = {"Authorization": "Bearer huggingface_api_token"}
+
+def query(payload):
+	response = requests.post(API_URL, headers=headers, json=payload)
+	return response.content
+
+image_bytes = query({
+	"inputs": "Astronaut riding a horse",
+})
+
+image = Image.open(io.BytesIO(image_bytes))
+image.save("generated_image.jpg")
+```
+or using Diffusers library (requires pytorch and transformers too):
+```Python
+from diffusers import DiffusionPipeline
+import io
+from PIL import *
+
+pipeline = DiffusionPipeline.from_pretrained("stablediffusionapi/juggernaut-xl-v5")
+pipeline.load_lora_weights("Kvikontent/kviimager2.0")
+
+prompt = "Astronaut riding a horse"
+
+image_bytes = pipeline(prompt)
+image = Image.open(io.BytesIO(image_bytes))
+image.save("generated_image.jpg")
+```
+
+## Credits
+
+* Author - Vasiliy Katsyka
+* Company - [KVIAI](https://hf.co/kviai)
+* Licence - Openrail
+
 ## Official demo
 
-You can use official demo on Spaces: [try](https://huggingface.co/spaces/kvikontent/kviimager).
+You can use official demo on Spaces: [try](https://huggingface.co/spaces/kvikontent/kviimager2.0).