felixrosberg commited on
Commit
e3c053a
·
verified ·
1 Parent(s): f656238

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ from engine.ui_model import fetch_model, run_model
17
  from engine.ui_gradio import fetch_ui
18
 
19
  engage_logo = Image.open("engage_studios_logo.png").resize((700, 88), Image.Resampling.BICUBIC)
20
- engage_logo_mask = np.array(engage_logo.split()[-1])[..., None] / 255
21
  engage_logo_np = np.array(engage_logo.convert('RGB'))
22
 
23
  pipe = fetch_model()
 
17
  from engine.ui_gradio import fetch_ui
18
 
19
  engage_logo = Image.open("engage_studios_logo.png").resize((700, 88), Image.Resampling.BICUBIC)
20
+ engage_logo_mask = np.clip(np.array(engage_logo.split()[-1])[..., None] / 255, 0, 1)
21
  engage_logo_np = np.array(engage_logo.convert('RGB'))
22
 
23
  pipe = fetch_model()