Senath commited on
Commit
c06020e
·
verified ·
1 Parent(s): 9a5c6b0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -10
app.py CHANGED
@@ -4,15 +4,12 @@ import torchaudio
4
  import gradio as gr
5
  from transformers import AutoProcessor, SeamlessM4TModel
6
 
7
- # Constants
8
  MODEL_NAME = "facebook/hf-seamless-m4t-medium"
9
  device = "cuda" if torch.cuda.is_available() else "cpu"
10
 
11
- # Load model and processor
12
  processor = AutoProcessor.from_pretrained(MODEL_NAME)
13
  model = SeamlessM4TModel.from_pretrained(MODEL_NAME).to(device).eval()
14
 
15
- # Main translation function
16
  def translate(text_input, audio_input, source_lang, target_lang, auto_detect):
17
  src = None if auto_detect else source_lang
18
  translated_text = None
@@ -38,8 +35,8 @@ def translate(text_input, audio_input, source_lang, target_lang, auto_detect):
38
  return translated_text or "", translated_audio
39
  return "No input provided.", None
40
 
41
- # ✅ This exposes the endpoint correctly
42
- gr.Interface(
43
  fn=translate,
44
  inputs=[
45
  gr.Textbox(label="Input Text (optional)"),
@@ -52,9 +49,9 @@ gr.Interface(
52
  gr.Textbox(label="Translated Text"),
53
  gr.Audio(label="Translated Speech")
54
  ],
55
- title="iVoice Translate (Text + Speech)"
56
- ).launch(
57
- server_name="0.0.0.0",
58
- server_port=int(os.environ.get("PORT", 7860)),
59
- show_api=True
60
  )
 
 
 
4
  import gradio as gr
5
  from transformers import AutoProcessor, SeamlessM4TModel
6
 
 
7
  MODEL_NAME = "facebook/hf-seamless-m4t-medium"
8
  device = "cuda" if torch.cuda.is_available() else "cpu"
9
 
 
10
  processor = AutoProcessor.from_pretrained(MODEL_NAME)
11
  model = SeamlessM4TModel.from_pretrained(MODEL_NAME).to(device).eval()
12
 
 
13
  def translate(text_input, audio_input, source_lang, target_lang, auto_detect):
14
  src = None if auto_detect else source_lang
15
  translated_text = None
 
35
  return translated_text or "", translated_audio
36
  return "No input provided.", None
37
 
38
+ # ✅ API name must be set here
39
+ demo = gr.Interface(
40
  fn=translate,
41
  inputs=[
42
  gr.Textbox(label="Input Text (optional)"),
 
49
  gr.Textbox(label="Translated Text"),
50
  gr.Audio(label="Translated Speech")
51
  ],
52
+ title="iVoice Translate (Text + Speech)",
53
+ # ✅ This exposes it at /predict
54
+ api_name="/predict"
 
 
55
  )
56
+
57
+ demo.launch(server_name="0.0.0.0", server_port=int(os.environ.get("PORT", 7860)), show_api=True)