Update app.py

app.py

@@ -9,17 +9,11 @@ from typing import Dict
 from PIL import Image
 from huggingface_hub import Repository
 
-token = os.environ['model_fetch']
-
-engine_repo = Repository(local_dir="engine", clone_from="felixrosberg/EngageDiffusion", use_auth_token=token)
+engine_repo = Repository(local_dir="engine", clone_from="felixrosberg/EngageDiffusion", use_auth_token=os.environ['model_fetch'])
 
 from engine.ui_model import fetch_model, run_model
 from engine.ui_gradio import fetch_ui
 
-engage_logo = Image.open("engage_studios_logo.png").resize((700, 88), Image.Resampling.BICUBIC)
-engage_logo_mask = np.clip(np.array(engage_logo.split()[-1])[..., None] / 255, 0, 1)
-engage_logo_np = np.array(engage_logo.convert('RGB'))
-
 pipe = fetch_model()
 pipe.to('cuda')
 
@@ -95,11 +89,6 @@ def inference(user_state, condition_image, settings, prompt, neg_prompt, inferen
               seed,
               progress)
 
-    for idx, im in enumerate(images):
-        im = np.asarray(im).copy()
-        im[-88:, :700] = im[-88:, :700] * (1 - engage_logo_mask) + engage_logo_np
-        images[idx] = Image.fromarray(np.clip(im.astype('uint8'), 0, 255))
-
     user_state["IMAGE_GALLERY"] += images
     return user_state["IMAGE_GALLERY"], user_state