Spaces:
Runtime error
Runtime error
wemab39501
commited on
Commit
•
e8b8fc7
1
Parent(s):
3eaed42
Update app.py
Browse files
app.py
CHANGED
@@ -21,9 +21,9 @@ device = "cuda"
|
|
21 |
#pipe = pipe.to(device)
|
22 |
#torch.backends.cudnn.benchmark = True
|
23 |
|
24 |
-
|
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):
|