AdamHider commited on
Commit
8823536
1 Parent(s): 9b71340
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -6,10 +6,10 @@ import torch
6
  model = VitsModel.from_pretrained("facebook/mms-tts-crh")
7
  tokenizer = AutoTokenizer.from_pretrained("facebook/mms-tts-crh")
8
 
9
- def greet(name):
10
  global model, tokenizer
11
  text = name
12
-
13
  inputs = tokenizer(text, return_tensors="pt")
14
 
15
  with torch.no_grad():
@@ -17,5 +17,9 @@ def greet(name):
17
 
18
  return output
19
 
20
- iface = gr.Interface(fn=greet, inputs="text", outputs="audio")
21
- iface.launch()
 
 
 
 
 
6
  model = VitsModel.from_pretrained("facebook/mms-tts-crh")
7
  tokenizer = AutoTokenizer.from_pretrained("facebook/mms-tts-crh")
8
 
9
+ def predict(image):
10
  global model, tokenizer
11
  text = name
12
+
13
  inputs = tokenizer(text, return_tensors="pt")
14
 
15
  with torch.no_grad():
 
17
 
18
  return output
19
 
20
+ gr.Interface(
21
+ predict,
22
+ inputs= gr.Textbox(value="", label="Duration in seconds"),,
23
+ outputs="audio",
24
+ title="Hot Dog? Or Not?",
25
+ ).launch()