Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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):
|