Hecheng0625
commited on
Commit
•
4cc2fd4
1
Parent(s):
386bae1
Update app.py
Browse files
app.py
CHANGED
@@ -19,7 +19,7 @@ from transformers import SeamlessM4TFeatureExtractor
|
|
19 |
|
20 |
processor = SeamlessM4TFeatureExtractor.from_pretrained("facebook/w2v-bert-2.0")
|
21 |
|
22 |
-
device = torch.device("cuda" if torch.cuda.is_available() else "CPU"
|
23 |
|
24 |
|
25 |
|
|
|
19 |
|
20 |
processor = SeamlessM4TFeatureExtractor.from_pretrained("facebook/w2v-bert-2.0")
|
21 |
|
22 |
+
device = torch.device("cuda" if torch.cuda.is_available() else "CPU")
|
23 |
|
24 |
|
25 |
|