Update app.py

app.py

@@ -22,8 +22,6 @@ from flax.jax_utils import replicate
 from flax.training.common_utils import shard
 
 os.environ["XLA_PYTHON_CLIENT_PREALLOCATE"]="false"
-os.environ["XLA_PYTHON_CLIENT_MEM_FRACTION"]=".XX"
-os.environ["XLA_PYTHON_CLIENT_ALLOCATOR"]="platform"
 
 def create_key(seed=0):
     return jax.random.PRNGKey(seed)
@@ -80,7 +78,7 @@ def infer(prompts, negative_prompts, image):
 title = "ControlNet + Stable Diffusion for Battlemaps"
 description = """Sketch your game battlemap and add some prompts to let the magic happen 🪄.
 Pretrained on battlemaps images. 
-By Orgrim, Karm and Robin "
+By Orgrim, Karm and Robin
 """
 # you need to pass inputs and outputs according to inference function
 gr.Interface(