lmattingly13 commited on
Commit
1dd294a
·
1 Parent(s): 26e526a

changed assignments, and updated card

Browse files
Files changed (3) hide show
  1. .gitignore +1 -0
  2. README.md +2 -0
  3. app.py +2 -1
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ __pycache__/app.cpython-39.pyc
README.md CHANGED
@@ -7,6 +7,8 @@ sdk: gradio
7
  sdk_version: 3.28.0
8
  app_file: app.py
9
  pinned: false
 
 
10
  tags:
11
  - jax-diffusers-event
12
  ---
 
7
  sdk_version: 3.28.0
8
  app_file: app.py
9
  pinned: false
10
+ models: ["runwayml/stable-diffusion-v1-5", "https://huggingface.co/lmattingly/controlnet-uncanny-simpsons"]
11
+ datasets: ["skiracer/simpsons_blip_captions","https://huggingface.co/datasets/lmattingly/simpsons_canny"]
12
  tags:
13
  - jax-diffusers-event
14
  ---
app.py CHANGED
@@ -51,7 +51,8 @@ def create_key(seed=0):
51
  def infer(prompts, image):
52
  params["controlnet"] = controlnet_params
53
 
54
- image = resize_image(image, 500)
 
55
  num_samples = 1 #jax.device_count()
56
  rng = create_key(0)
57
  rng = jax.random.split(rng, jax.device_count())
 
51
  def infer(prompts, image):
52
  params["controlnet"] = controlnet_params
53
 
54
+ im = image
55
+ image = resize_image(im, 500)
56
  num_samples = 1 #jax.device_count()
57
  rng = create_key(0)
58
  rng = jax.random.split(rng, jax.device_count())