Update app.py

app.py

@@ -4,7 +4,6 @@ import uuid
 from urllib.parse import quote
 from requests import get
 from bs4 import BeautifulSoup
-import hashlib
 
 import gradio as gr
 import numpy as np
@@ -79,8 +78,6 @@ def translate_to_english(phrase, src_lang):
 
 
 @spaces.GPU(enable_queue=True)
-CACHE_DIR = "image_cache"
-
 def generate(
     phrase: str,
     input_lang: str,
@@ -91,6 +88,7 @@ def generate(
     height: int = 1024,
     guidance_scale: float = 3,
     randomize_seed: bool = False,
+    use_resolution_binning: bool = True,
     progress=gr.Progress(track_tqdm=True),
 ):
     pipe.to(device)
@@ -105,14 +103,6 @@ def generate(
     if not use_negative_prompt:
         negative_prompt = None  # type: ignore
 
-    # Hash the prompt and seed to create a unique cache key
-    cache_key = hashlib.sha256(f"{prompt}{seed}".encode()).hexdigest()
-
-    # Check if the image is already in the cache
-    if os.path.exists(os.path.join(CACHE_DIR, f"{cache_key}.png")):
-        print("Image found in cache")
-        return [os.path.join(CACHE_DIR, f"{cache_key}.png")], seed
-
     images = pipe(
         prompt=prompt,
         negative_prompt=negative_prompt,
@@ -128,13 +118,6 @@ def generate(
 
     image_paths = [save_image(img) for img in images]
     print(image_paths)
-
-    # Save the first image to the cache directory
-    if not os.path.exists(CACHE_DIR):
-        os.makedirs(CACHE_DIR)
-    cache_path = os.path.join(CACHE_DIR, f"{cache_key}.png")
-    Image.open(image_paths[0]).save(cache_path)
-
     return image_paths, seed