10

app.py

@@ -1,6 +1,6 @@
 import os
-os.environ['CUDA_DEVICE_ORDER'] = 'PCI_BUS_ID'
-os.environ['CUDA_VISIBLE_DEVICES'] = '0'
+# os.environ['CUDA_DEVICE_ORDER'] = 'PCI_BUS_ID'
+# os.environ['CUDA_VISIBLE_DEVICES'] = '2'
 import gradio as gr
 import numpy as np
 from models import make_inpainting
@@ -85,7 +85,7 @@ def upscale(image, prompt):
 
     ret = pipeline(prompt=prompt, 
                    image=image,
-                   num_inference_steps=20,
+                   num_inference_steps=10,
                    guidance_scale=0)
     print("ret",ret)
     upscaled_image = ret.images[0]