Update README.md

README.md

@@ -39,13 +39,38 @@ This project involves a text-to-image model fine-tuned on the Stable Diffusion 1
 
 ## Usage
 To generate an image:
-```python
-from model import NFTGenesisModel
+```import requests
+import io
+from PIL import Image
+
+API_URL = "https://api-inference.huggingface.co/models/sarathAI/NFT-Genesis"
+headers = {"Authorization": "Bearer XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX"}
+
+def query(payload):
+    response = requests.post(API_URL, headers=headers, json=payload)
+    return response.content
+
+image_bytes = query({
+    "inputs": "Formula 1 car",
+})
+
+# Added: Check if the response is indeed image bytes
+if image_bytes.startswith(b'\xff\xd8'):  # JPEG
+    print("JPEG image detected")
+elif image_bytes.startswith(b'\x89PNG\r\n\x1a\n'):  # PNG
+    print("PNG image detected")
+else:
+    print("The response might not be an image or is in an unrecognized format.")
+
+# Attempt to open the image
+try:
+    image = Image.open(io.BytesIO(image_bytes))
+    image.save("output_image.jpg")
+    print("Image saved as output_image.jpg. Please open this file to view the image.")
+except IOError:
+    print("Cannot open the image. The file might be corrupted or in an unsupported format.")
+
 
-model = NFTGenesisModel()
-description = "Your textual description here"
-image = model.generate(description)
-image.save("output.jpg")
 ```
 
 ## Configuration