sec

app.py

@@ -15,11 +15,8 @@ import numpy as np
 import os 
 import shutil
 warnings.filterwarnings("ignore")
-
 st.title('Background Generation')
-
 st.write('This app generates new backgrounds for images.')
-
 # set environment variable for dll
 os.environ['KMP_DUPLICATE_LIB_OK']='True'
 
@@ -148,9 +145,9 @@ run_model_button = st.button("Run Model")
 def initialize_pipe():
     pipe = StableDiffusionInpaintPipeline.from_pretrained("runwayml/stable-diffusion-inpainting",
             revision="fp16",
-            torch_dtype=torch.float16,
+            torch_dtype=torch.float32, #16 for gpu
             safety_checker = None,
-            requires_safety_checker = False).to("cuda")
+            requires_safety_checker = False) #.to("cuda")
 
     pipe.safety_checker = None
     pipe.requires_safety_checker = False