Update app.py

app.py

@@ -1,5 +1,6 @@
 from diffusers import AutoPipelineForText2Image, StableDiffusionImg2ImgPipeline
 import torchvision.transforms.functional as fn
+import torchvision.transforms.InterpolationMode as interp
 import gradio as gr
 import random
 import torch
@@ -25,7 +26,7 @@ def generate(prompt, samp_steps, batch_size, seed, progress=gr.Progress(track_tq
         guidance_scale=0.0,
         generator=torch.manual_seed(seed),
     ).images
-    upscaled_images = fn.resize(images, 1024, InterpolationMode.NEAREST_EXACT)
+    upscaled_images = fn.resize(images, 1024, interp.NEAREST_EXACT)
     final_images = img2img(
         prompt,
         num_inference_steps=samp_steps,