Update app.py

app.py

@@ -3,12 +3,18 @@ import numpy as np
 import gradio as gr
 import spaces
 import cv2
+import os
 
 from typing import Dict
 from torchvision.transforms.functional import to_tensor, center_crop, resize
 from PIL import Image
+from huggingface_hub import Repository
 
-from ui_model import fetch_model, process_sketch, process_mask
+token = os.environ['model_fetch']
+
+engine_repo = Repository(local_dir="engine", clone_from="felixrosberg/EngageDiffusion", use_auth_token=token)
+
+from engine.ui_model import fetch_model, process_sketch, process_mask
 
 engage_logo = Image.open("engage_studios_logo.png").resize((700, 88), Image.Resampling.BICUBIC)
 engage_logo_mask = np.array(engage_logo.split()[-1])[..., None] / 255