push

main.py

@@ -1,15 +1,11 @@
 import os
-import io
-import torch
 from flask import Flask, request, jsonify, send_file
 from flask_cors import CORS
 from diffusers import AutoPipelineForImage2Image
 from diffusers.utils import load_image, make_image_grid
 from PIL import Image
-
-# Set the cache directory to a location with write permissions
-os.environ['TRANSFORMERS_CACHE'] = './transformers_cache'
-os.environ['HF_HOME'] = './transformers_cache'
+import torch
+import io
 
 app = Flask(__name__)
 CORS(app)