Update app.py

app.py

@@ -31,7 +31,7 @@ from models.pipeline_consisid import ConsisIDPipeline
 from models.eva_clip import create_model_and_transforms
 from models.eva_clip.constants import OPENAI_DATASET_MEAN, OPENAI_DATASET_STD
 from models.eva_clip.utils_qformer import resize_numpy_image_long
-
+os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 hf_hub_download(repo_id="ai-forever/Real-ESRGAN", filename="RealESRGAN_x4.pth", local_dir="model_real_esran")