Update app.py

app.py

@@ -14,6 +14,7 @@ from diffusers.models.attention_processor import Attention
 from transformers import AutoProcessor, AutoModelForMaskGeneration, pipeline
 from dataclasses import dataclass
 from typing import Any, List, Dict, Optional, Union, Tuple
+from huggingface_hub import hf_hub_download
 import random 
 
 device = "cuda" if torch.cuda.is_available() else "cpu"