angtrim commited on
Commit
5fadb62
1 Parent(s): fb8291f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -35,7 +35,7 @@ def run(image, item_name):
35
  ones = Image.new("L", (512, 512), 255)
36
  temp_uri = pil_image_to_data_uri(ones)
37
  return ones, gr.update(link=temp_uri), gr.update(link=temp_uri)
38
- prompt = item_name + "professional 3d model. octane render, highly detailed, volumetric, dramatic lighting"
39
  image = image.convert("RGB")
40
  image_t = TF.to_tensor(image) > 0.5
41
  image_pil = TF.to_pil_image(image_t.to(torch.float32))
@@ -56,7 +56,7 @@ def update_canvas(use_line, use_eraser):
56
  _color = "#ffffff"
57
  brush_size = 20
58
  if use_line:
59
- _color = "#000000"
60
  brush_size = 4
61
  return gr.update(brush_radius=brush_size, brush_color=_color, interactive=True)
62
 
 
35
  ones = Image.new("L", (512, 512), 255)
36
  temp_uri = pil_image_to_data_uri(ones)
37
  return ones, gr.update(link=temp_uri), gr.update(link=temp_uri)
38
+ prompt = item_name + " professional 3d model. octane render, highly detailed, volumetric, dramatic lighting"
39
  image = image.convert("RGB")
40
  image_t = TF.to_tensor(image) > 0.5
41
  image_pil = TF.to_pil_image(image_t.to(torch.float32))
 
56
  _color = "#ffffff"
57
  brush_size = 20
58
  if use_line:
59
+ _color = "#000011"
60
  brush_size = 4
61
  return gr.update(brush_radius=brush_size, brush_color=_color, interactive=True)
62