update app.py

app.py

@@ -5,6 +5,9 @@ warnings.filterwarnings('ignore')
 import subprocess, io, os, sys, time
 from loguru import logger
 
+os.system("pip install diffuser==0.16.0")
+os.system("pip install transformers==4.29.1")
+
 os.environ["CUDA_VISIBLE_DEVICES"] = "0"
 
 if os.environ.get('IS_MY_DEBUG') is None:
@@ -241,12 +244,14 @@ sam_model = build_sam(checkpoint=sam_checkpoint) # .to(device)
 sam_predictor = SamPredictor(sam_model)
 sam_mask_generator = SamAutomaticMaskGenerator(sam_model)
 
+os.system("pip list")
 # initialize stable-diffusion-inpainting
 logger.info(f"initialize stable-diffusion-inpainting...")
 sd_pipe = None
 if os.environ.get('IS_MY_DEBUG') is None:
     sd_pipe = StableDiffusionInpaintPipeline.from_pretrained(
             # "runwayml/stable-diffusion-inpainting", 
+            # revision="fp16"
             "stabilityai/stable-diffusion-2-inpainting",
             torch_dtype=torch.float16
     )

requirements.txt

@@ -1,6 +1,6 @@
 accelerate
 addict
-diffusers
+diffusers==0.16.0
 gradio
 huggingface_hub
 matplotlib
@@ -17,11 +17,12 @@ termcolor
 timm
 torch
 torchvision
-transformers
+transformers==4.29.1
 yapf
 numba
-# segment_anything
+scipy 
+safetensors
 
 lama-cleaner==0.25.0
 openmim==0.1.5
-mmcv==2.0.0
+mmcv==2.0.0