Add application file

app.py

@@ -0,0 +1,40 @@
+import torch
+import os
+import gradio as gr
+import numpy as np
+import urllib.request
+from PIL import Image
+
+# Tải mô hình GAN đã lưu
+model_url = "https://github.com/NguyenLe2004/cat-image-generate/raw/refs/heads/main/generate/cat_face_generate_model.pt" 
+model_path = "cat_image_generate.pt"
+
+# Tải mô hình nếu chưa tồn tại
+if not os.path.exists(model_path):
+    urllib.request.urlretrieve(model_url, model_path)
+
+# Load mô hình
+model = torch.jit.load(model_path, map_location=torch.device('cpu'))
+model.eval()
+
+def generate_image():
+    # Tạo một vector ngẫu nhiên
+    latent_dim = 100 
+    z = torch.randn(1, latent_dim, 1 , 1) 
+    
+    with torch.no_grad():
+        generated_image = model(z)
+    generated_image = generated_image.squeeze().numpy()
+    generated_image = generated_image*0.5 + 0.5
+    generated_image = np.transpose(generated_image, (1, 2, 0))  
+
+    # Chuyển về kiểu uint8 để hiển thị
+    generated_image = (generated_image * 255).clip(0, 255).astype(np.uint8)
+
+    # Dùng PIL để chuyển thành ảnh
+    image = Image.fromarray(generated_image)
+    return image
+
+# Giao diện Gradio
+demo = gr.Interface(fn=generate_image, inputs=None, outputs="image")
+demo.launch()