Changes

app.py

@@ -11,13 +11,13 @@ from huggingface_hub import login
 
 # HF_TOKEN = os.environ.get("HF_TOKEN")
 
-login()
+login(token='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')
 
 
 # model_path = WEIGHTS_DIR             # If you want to use previously trained model saved in gdrive, replace this with the full path of model in gdrive
 # headers = {"Authorization": "Bearer xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"}
 
-pipe = StableDiffusionPipeline.from_pretrained("Divyanshu04/Finetuned-model", safety_checker=None, torch_dtype=torch.float32).to("cuda")
+pipe = StableDiffusionPipeline.from_pretrained("Divyanshu04/Finetuned-model", safety_checker=None, torch_dtype=torch.float32).to("cpu")
 
 pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
 pipe.enable_xformers_memory_efficient_attention()
@@ -49,13 +49,13 @@ def generate():
         height = 512
         width = 512
 
-        g_cuda = torch.Generator(device='cuda')
+        g_cuda = torch.Generator(device='cpu')
         seed = 52362
         g_cuda.manual_seed(seed)
 
         # commandline_args = os.environ.get('COMMANDLINE_ARGS', "--skip-torch-cuda-test --no-half")
 
-        with autocast("cuda"), torch.inference_mode():
+        with autocast("cpu"), torch.inference_mode():
             images = pipe(
                 prompt,
                 height=height,