Update app.py

app.py

@@ -1,9 +1,5 @@
 import gradio as gr
-
-import requests
 from PIL import Image
-from pathlib import Path
-from io import BytesIO
 
 # Diffusers
 from diffusers import (
@@ -11,18 +7,14 @@ from diffusers import (
     FlaxStableDiffusionControlNetPipeline
 )
 from diffusers.utils import load_image
-
-# Pytorch
+# PyTorch
 import torch
-
 # Numpy
 import numpy as np
-
 # Jax
 import jax
 import jax.numpy as jnp
 from jax import pmap
-
 # Flax
 import flax
 from flax.jax_utils import replicate
@@ -53,11 +45,11 @@ def infer(prompts, negative_prompts, image):
     num_samples = 1  # jax.device_count()
     rng = create_key(0)
     rng = jax.random.split(rng, jax.device_count())
-    #battlemap_image = Image.open(image)
+    battlemap_image = Image.open(image).copy()
 
     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) #battlemap_image
+    processed_image = pipe.prepare_image_inputs([battlemap_image] * num_samples) #battlemap_image
 
     p_params = replicate(params)
     prompt_ids = shard(prompt_ids)