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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,7 +7,7 @@ import numpy as np
7
  import torch
8
  import torchaudio
9
  import spaces
10
- from transformers import AutoProcessor, SeamlessM4TModel
11
 
12
  from lang_list import (
13
  LANGUAGE_NAME_TO_CODE,
@@ -41,7 +41,7 @@ DEFAULT_TARGET_LANGUAGE = "French"
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
 
 
7
  import torch
8
  import torchaudio
9
  import spaces
10
+ from transformers import AutoProcessor, SeamlessM4Tv2Model
11
 
12
  from lang_list import (
13
  LANGUAGE_NAME_TO_CODE,
 
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 = SeamlessM4Tv2Model.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