Initial_commit

app.py

@@ -1,6 +1,6 @@
-import os
-import pdb
-from PIL import Image
+# import os
+# import pdb
+# from PIL import Image
 import gradio as gr
 
 from src.utils.gradio_utils import *

src/utils/gradio_utils.py

@@ -1,4 +1,4 @@
-import numpy
+import numpy as np
 from tqdm import tqdm
 import gradio as gr
 import os 
@@ -8,12 +8,12 @@ import torch.nn.functional as F
 import torch.utils.checkpoint
 
 from accelerate import Accelerator
-from accelerate.logging import get_logger
+# from accelerate.logging import get_logger
 from accelerate.utils import set_seed
 from diffusers import AutoencoderKL, DDPMScheduler, DDIMScheduler, UNet2DConditionModel
-from huggingface_hub import HfFolder, Repository, whoami
+# from huggingface_hub import HfFolder, Repository, whoami
 from PIL import Image
-import numpy as np
+
 from torchvision import transforms
 from transformers import CLIPTextModel, CLIPTokenizer
 from torch import autocast