try fix memory

app.py

@@ -1,6 +1,7 @@
 import os
 # os.environ['CUDA_DEVICE_ORDER'] = 'PCI_BUS_ID'
 # os.environ['CUDA_VISIBLE_DEVICES'] = '2'
+os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "caching_allocator"
 import gradio as gr
 import numpy as np
 from models import make_inpainting
@@ -93,6 +94,7 @@ def upscale(image, prompt):
 
     return upscaled_image
 
+
 with gr.Blocks() as app:    
     with gr.Row():
 
@@ -123,6 +125,7 @@ with gr.Blocks() as app:
         with gr.Column():
             gr.Button("Upscale").click(upscale, inputs=[gr.Image(type="pil"),gr.Textbox(label="prompt",value="empty room")], outputs=gr.Image())
 
+
 app.launch(debug=True,share=True)
 
 # UP 1