wemab39501 commited on
Commit
e8b8fc7
1 Parent(s): 3eaed42

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -21,9 +21,9 @@ device = "cuda"
21
  #pipe = pipe.to(device)
22
  #torch.backends.cudnn.benchmark = True
23
 
24
- #When running locally, you won`t have access to this, so you can remove this part
25
- word_list_dataset = load_dataset("stabilityai/word-list", data_files="list.txt", use_auth_token=True)
26
- word_list = word_list_dataset["train"]['text']
27
 
28
  is_gpu_busy = False
29
  def infer(prompt):
 
21
  #pipe = pipe.to(device)
22
  #torch.backends.cudnn.benchmark = True
23
 
24
+ ##When running locally, you won`t have access to this, so you can remove this part
25
+ #word_list_dataset = load_dataset("stabilityai/word-list", data_files="list.txt", use_auth_token=True)
26
+ #word_list = word_list_dataset["train"]['text']
27
 
28
  is_gpu_busy = False
29
  def infer(prompt):