update: application + dependencies

app.py

@@ -1,4 +1,3 @@
-import os
 import shutil
 
 import gradio as gr
@@ -28,7 +27,7 @@ def generate_image(
     seed,
 ):
     if not (wandb_api_key is None or wandb_api_key == ""):
-        os.environ["WANDB_API_KEY"] = wandb_api_key
+        wandb.login(key=wandb_api_key, relogin=True)
         generator = torch.Generator(device="cuda").manual_seed(seed)
         autolog(init=dict(project=wandb_project))
         run_url = wandb.run.get_url()
@@ -41,6 +40,7 @@ def generate_image(
             guidance_scale=guidance_scale,
             num_inference_steps=num_inference_steps,
         ).images[0]
+        wandb.finish()
         if torch.cuda.is_available():
             torch.cuda.empty_cache()
         shutil.rmtree("wandb")

requirements.txt

@@ -5,4 +5,4 @@ accelerate
 torch
 ftfy
 sentencepiece
-wandb @ git+https://github.com/wandb/wandb@main
+wandb