Update app.py

app.py

@@ -2,7 +2,14 @@ import gradio as gr
 import torch
 from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
 
+import numpy as np
+import os
+import cv2
+from PIL import Image, ImageDraw
+import insightface
+from insightface.app import FaceAnalysis
 
+# Diffusion
 model_base = "runwayml/stable-diffusion-v1-5"
 
 pipe = StableDiffusionPipeline.from_pretrained(model_base, torch_dtype=torch.float16, use_safetensors=True, safety_checker=None,)
@@ -13,6 +20,12 @@ pipe.unet.load_attn_procs(lora_model_path)
 pipe.to("cuda")
 
 
+# Insightface model
+app = FaceAnalysis(name='buffalo_l')
+app.prepare(ctx_id=0, det_size=(640, 640))
+
+
+
 def greet(description,color,features,occasion,type_):
     
     prompt = 'white background '