Update app.py

app.py

@@ -25,12 +25,12 @@ def infer(prompt):
     num_samples = 1 #jax.device_count()
     rng = create_key(0)
     rng = jax.random.split(rng, jax.device_count())
-    im = image
-    image = Image.fromarray(im)
+    #im = image
+    #image = Image.fromarray(im)
     
     prompt_ids = pipe.prepare_text_inputs([prompts] * num_samples)
     #negative_prompt_ids = pipe.prepare_text_inputs([negative_prompts] * num_samples)
-    processed_image = pipe.prepare_image_inputs([image] * num_samples)
+    #processed_image = pipe.prepare_image_inputs([image] * num_samples)
     
     p_params = replicate(params)
     prompt_ids = shard(prompt_ids)