disable cache for now

app.py

@@ -24,8 +24,8 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
 
 
 def generate(
-    input_image: Image,
     prompt: str,
+    input_image: Image,
     negative_prompt: str = "",
     seed: int = 0,
     width: int = 1024,
@@ -39,6 +39,7 @@ def generate(
     num_images_per_prompt: int = 2,
 
 ) -> Image:
+
     payload = {
         "input": {
             "hdr": 0,
@@ -51,18 +52,22 @@ def generate(
             "resolution": "original",
             "resemblance": 0.75,
             "guidance_scale": 7,
-            "negative_prompt": "teeth, tooth, open mouth, longbody, lowres, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, mutant"
+            "negative_prompt": negative_prompt,
         }
     }
+    print("\n\n PAYLOAD", payload)
     response = requests.post(URL, headers=HEADERS, json=payload)
     json_response = response.json()
-    if 'output' in json_response:
+    if 'status' in json_response:
+        if json_response["status"] == "failed":
+            raise gr.Error("Failed to generate image")
         base64_image = json_response["output"][0]
         image_data = base64.b64decode(
             base64_image.replace("data:image/png;base64,", ""))
         image_stream = io.BytesIO(image_data)
         return [Image.open(input_image), Image.open(image_stream)]
-    raise gr.Error(json_response["status"])
+    else:
+        raise gr.Error("Failed to generate image")
 
 
 examples = [
@@ -162,12 +167,12 @@ with gr.Blocks() as demo:
         inputs=[prompt, input_image],
         outputs=result,
         fn=generate,
-        cache_examples=True,
+        cache_examples=False,
     )
 
     inputs = [
-        input_image,
         prompt,
+        input_image,
         negative_prompt,
         seed,
         width,

run.sh

@@ -1,2 +1,2 @@
-cd /src && python3 -m cog.server.http --threads=10 &
+cd /src && python3 -m cog.server.http --threads=10 & sleep 10s &&
 cd $HOME/app && . $HOME/.venv/bin/activate && python app.py