Create generate_images.py

generate_images.py

@@ -0,0 +1,26 @@
+from transformers import AutoModelForGenerativePreTraining
+import numpy as np
+
+# Tải model từ file checkpoint
+model = AutoModelForGenerativePreTraining.from_pretrained("path/to/your/checkpoint/file.ckpt")
+
+# Mã hóa đầu vào văn bản
+input_text = "Tạo một bức tranh đẹp về một cảnh hoàng hôn."
+encoded_input = tokenizer.encode(input_text, return_tensors="pt")
+
+# Tạo ảnh
+num_images = 1
+generated_images = model.generate(encoded_input, num_images=num_images)
+
+# Chuyển đổi đầu ra thành mảng numpy
+generated_images = generated_images.numpy()
+
+# Lưu ảnh
+for i, image in enumerate(generated_images):
+    image = (image * 255).astype(np.uint8)
+    image = image.transpose(1, 2, 0)  # Chuyển đổi hình thức của ảnh
+    os.makedirs("images", exist_ok=True)
+    np.savez("images/image_{}.npz".format(i), image)
+    print(f"Ảnh {i} đã được lưu.")
+
+print("Tất cả ảnh đã được tạo và lưu thành công.")