Update app.py

app.py

@@ -1,13 +1,13 @@
 import os
-from flask import Flask, request, jsonify
+from flask import Flask, request, jsonify, send_file
 from PIL import Image
 from io import BytesIO
 import base64
 import torch
-import spaces
 import requests
 import numpy as np
 import uuid
+import spaces
 from transformers import (
     CLIPImageProcessor,
     CLIPVisionModelWithProjection,
@@ -15,7 +15,7 @@ from transformers import (
     CLIPTextModelWithProjection,
     AutoTokenizer
 )
-from diffusers import DDPMScheduler, AutoencoderKL
+from diffusers import DDPMScheduler, AutoencoderKL, UNet2DConditionModel
 from utils_mask import get_mask_location
 from torchvision import transforms
 import apply_net