Update app.py

app.py

@@ -359,8 +359,6 @@ def infer_compo(style_description, ref_style_file, caption, ref_sub_file, progre
         return sampled_image  # Return the sampled_image PIL image
 
     finally:
-        models_to(sam_model, device="cpu")
-        models_to(sam_model.sam, device="cpu")
         # Clear CUDA cache
         torch.cuda.empty_cache()
         gc.collect()