Update app.py

app.py

@@ -4,6 +4,7 @@ import torch
 import argparse
 import numpy as np
 import gradio as gr
+import tempfile
 
 from PIL import Image
 from copy import deepcopy
@@ -90,6 +91,8 @@ model = model.to(device)
 print ("IMAGE MODEL CKPT:", MODEL_NAME)
 load_network(model, MODEL_NAME, strict=True, param_key='params')
 
+# Create a temporary directory for Gradio
+temp_dir = tempfile.TemporaryDirectory()
 
 css = """
    /* .image-frame img, .image-container img {
@@ -102,13 +105,13 @@ css = """
 
 demo = gr.Interface(
     fn=process_img,
-    inputs=[gr.Image(type="pil", label="Изображение"),],
+    inputs=[gr.Image(type="pil", label="Изображение", temp_dir=temp_dir.name)],
     outputs=ImageSlider(label="Улучшенное качество", 
                         type="pil",
                         show_download_button=True,
-                        ), #[gr.Image(type="pil", label="Улучшеное изображение", min_width=500),],
+                        ),
     css=css,
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()