Update app.py

app.py

@@ -1,17 +1,18 @@
 import gradio as gr
-# import torch
-# from torch import autocast
-# from diffusers import StableDiffusionPipeline
-# from datasets import load_dataset
+import torch
+#from torch import autocast
+from diffusers import StableDiffusionPipeline
+#from datasets import load_dataset
 from PIL import Image  
 import re
 import os
 
-# auth_token = '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' #os.getenv("auth_token")
-# model_id = "CompVis/stable-diffusion-v1-4"
-# device = "cpu"
-# pipe = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token=auth_token, revision="fp16", torch_dtype=torch.float16)
-# pipe = pipe.to(device)
+auth_token = os.getenv("auth_token")
+model_id = "CompVis/stable-diffusion-v1-4"
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu") #device = "cuda"
+print(device)
+pipe = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token=auth_token, revision="fp16", torch_dtype=torch.float32)
+pipe = pipe.to(device)
 
 stable_diffusion = gr.Blocks.load(name="spaces/stabilityai/stable-diffusion")