Shaleen123 commited on
Commit
93f70f6
·
verified ·
1 Parent(s): 083b9ef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -3,14 +3,14 @@ import numpy as np
3
  import random
4
  from diffusers import DiffusionPipeline
5
  import torch
6
- from huggingface_hub import notebook_login
7
 
8
  device = "cuda" if torch.cuda.is_available() else "cpu"
9
 
10
- import os
11
- token_key = os.getenv('HF_KEY')
12
 
13
- notebook_login(token_key)
14
 
15
  if torch.cuda.is_available():
16
  torch.cuda.max_memory_allocated(device=device)
 
3
  import random
4
  from diffusers import DiffusionPipeline
5
  import torch
6
+ # from huggingface_hub import notebook_login
7
 
8
  device = "cuda" if torch.cuda.is_available() else "cpu"
9
 
10
+ # import os
11
+ # token_key = os.getenv('HF_KEY')
12
 
13
+ # # notebook_login(token_key)
14
 
15
  if torch.cuda.is_available():
16
  torch.cuda.max_memory_allocated(device=device)