Update

app_stylization.py

@@ -7,7 +7,7 @@ import torch
 from controlnet_aux import CannyDetector
 from diffusers.pipelines import BlipDiffusionControlNetPipeline
 
-from settings import CACHE_EXAMPLES, DEFAULT_NEGATIVE_PROMPT, MAX_INFERENCE_STEPS
+from settings import DEFAULT_NEGATIVE_PROMPT, MAX_INFERENCE_STEPS
 from utils import MAX_SEED, randomize_seed_fn
 
 canny_detector = CannyDetector()
@@ -105,7 +105,6 @@ with gr.Blocks() as demo:
         ],
         outputs=result,
         fn=run,
-        cache_examples=CACHE_EXAMPLES,
     )
 
     inputs = [

app_zero_shot.py

@@ -6,14 +6,12 @@ import spaces
 import torch
 from diffusers.pipelines import BlipDiffusionPipeline
 
-from settings import CACHE_EXAMPLES, DEFAULT_NEGATIVE_PROMPT, MAX_INFERENCE_STEPS
+from settings import DEFAULT_NEGATIVE_PROMPT, MAX_INFERENCE_STEPS
 from utils import MAX_SEED, randomize_seed_fn
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 if torch.cuda.is_available():
     pipe = BlipDiffusionPipeline.from_pretrained("Salesforce/blipdiffusion", torch_dtype=torch.float16).to(device)
-else:
-    pipe = None
 
 
 @spaces.GPU
@@ -95,7 +93,6 @@ with gr.Blocks() as demo:
         ],
         outputs=result,
         fn=run,
-        cache_examples=CACHE_EXAMPLES,
     )
 
     inputs = [

settings.py

@@ -1,5 +1,2 @@
-import os
-
 MAX_INFERENCE_STEPS = 50
 DEFAULT_NEGATIVE_PROMPT = "over-exposure, under-exposure, saturated, duplicate, out of frame, lowres, cropped, worst quality, low quality, jpeg artifacts, morbid, mutilated, out of frame, ugly, bad anatomy, bad proportions, deformed, blurry, duplicate"
-CACHE_EXAMPLES = os.getenv("CACHE_EXAMPLES") == "1"