lihaoxin2020 commited on
Commit
fb74c5a
·
verified ·
1 Parent(s): b05ad76

Update model

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -40,8 +40,8 @@ DEFAULT_TARGET_LANGUAGE = "French"
40
 
41
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
42
 
43
- processor = AutoProcessor.from_pretrained("ylacombe/hf-seamless-m4t-large")
44
- model = SeamlessM4TModel.from_pretrained("ylacombe/hf-seamless-m4t-large").to(device)
45
  # processor = AutoProcessor.from_pretrained("facebook/hf-seamless-m4t-medium")
46
  # model = SeamlessM4TModel.from_pretrained("facebook/hf-seamless-m4t-medium").to(device)
47
 
 
40
 
41
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
42
 
43
+ processor = AutoProcessor.from_pretrained("facebook/seamless-m4t-v2-large")
44
+ model = SeamlessM4TModel.from_pretrained("facebook/seamless-m4t-v2-large").to(device)
45
  # processor = AutoProcessor.from_pretrained("facebook/hf-seamless-m4t-medium")
46
  # model = SeamlessM4TModel.from_pretrained("facebook/hf-seamless-m4t-medium").to(device)
47