Update app.py

app.py

@@ -2,15 +2,17 @@ import gradio as gr
 import json
 import logging
 import torch
-from PIL import Image
+from PIL import Image, PngImagePlugin
 import spaces
 from diffusers import DiffusionPipeline
+from transformers.utils.hub import move_cache
 import copy
 import random
-import time
+import os
 import pygsheets
+import time
+
 
-from transformers.utils.hub import move_cache
 
 # Move cache
 move_cache()