Upload new_finetune.py

new_finetune.py

@@ -0,0 +1,46 @@
+# -*- coding: utf-8 -*-
+"""New_Finetune.ipynb
+
+Automatically generated by Colaboratory.
+
+Original file is located at
+    https://colab.research.google.com/drive/1HBmhGB58E7LDJPmp4fGI-mJhRb5OZDWo
+"""
+
+!pip install huggingface_hub
+from huggingface_hub import notebook_login
+notebook_login()
+
+!pip install --upgrade diffusers[torch]
+
+from torch import autocast
+from diffusers import StableDiffusionPipeline
+
+pipe = StableDiffusionPipeline.from_pretrained(
+    "CompVis/stable-diffusion-v1-4",
+    use_auth_token=True
+).to("cuda")
+
+prompt = "a sleek logo of a company, named 'SPARK', emblem style, our moto is 'Sparking some code to win', abstract logo, professinal, centre align, black background, the logo should resemble the name, clean"
+with autocast("cuda"):
+    output = pipe(prompt)
+print(output.keys())
+
+import matplotlib.pyplot as plt
+import numpy as np
+image_data = output["images"]
+
+# If the image data is a list, assume it contains PIL images
+if isinstance(image_data, list):
+    for i, image in enumerate(image_data):
+        # Convert PIL image to numpy array
+        image_np = np.array(image)
+
+        # Display the image using Matplotlib
+        plt.imshow(image_np)
+        plt.axis('off')
+        plt.title(f"Image {i+1}")
+        plt.show()
+else:
+    print("Unexpected image format. Unable to display.")
+