Update app.py

app.py

@@ -9,7 +9,12 @@ from PIL import Image
 import numpy as np
 from io import BytesIO
 import os
-
+import torch
+from torch import autocast
+from diffusers import StableDiffusionPipeline
+from datasets import load_dataset
+from PIL import Image  
+import re
 
 source_img = gr.Image(source="upload", type="filepath", label="init_img | 512*512 px")
 gallery = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery").style(grid=[2], height="auto")