Shenuki commited on
Commit
f27eac9
·
verified ·
1 Parent(s): c7be9a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,12 +1,12 @@
1
  import os
2
  import torch
3
- from transformers import AutoProcessor, SeamlessM4TForTextToText
4
  import gradio as gr
5
 
6
  MODEL_NAME = "facebook/hf-seamless-m4t-medium"
7
  device = "cuda" if torch.cuda.is_available() else "cpu"
8
 
9
- processor = AutoProcessor.from_pretrained(MODEL_NAME)
10
  model = SeamlessM4TForTextToText.from_pretrained(MODEL_NAME).to(device).eval()
11
 
12
  def translate(text, sourceLang, targetLang, auto_detect):
 
1
  import os
2
  import torch
3
+ from transformers import AutoProcessor, SeamlessM4TForTextToText, SeamlessM4TProcessor
4
  import gradio as gr
5
 
6
  MODEL_NAME = "facebook/hf-seamless-m4t-medium"
7
  device = "cuda" if torch.cuda.is_available() else "cpu"
8
 
9
+ processor = SeamlessM4TProcessor.from_pretrained(MODEL_NAME)
10
  model = SeamlessM4TForTextToText.from_pretrained(MODEL_NAME).to(device).eval()
11
 
12
  def translate(text, sourceLang, targetLang, auto_detect):