Francesco commited on
Commit
e34b1b5
1 Parent(s): c8030d6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -22,8 +22,8 @@ device = "cuda"
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):
 
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):