Spaces:

gaur3009
/

Imagier2

Sleeping

gaur3009 commited on Jul 23, 2024

Commit

b058a84

verified ·

1 Parent(s): 836c032

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ from diffusers.utils import load_image
 import math
 device = "cuda" if torch.cuda.is_available() else "cpu"
-pipe = AutoPipelineForImage2Image.from_pretrained("radames/stable-diffusion-v1-5-img2img", torch_dtype=torch.float16) if torch.cuda.is_available() else AutoPipelineForImage2Image.from_pretrained("radames/stable-diffusion-v1-5-img2img")
 pipe = pipe.to(device)
 def resize(value,img):

 import math
 device = "cuda" if torch.cuda.is_available() else "cpu"
+pipe = AutoPipelineForImage2Image.from_pretrained("openai/shap-e-img2img", torch_dtype=torch.float16) if torch.cuda.is_available() else AutoPipelineForImage2Image.from_pretrained("openai/shap-e-img2img")
 pipe = pipe.to(device)
 def resize(value,img):