Update app.py

app.py

@@ -31,13 +31,20 @@ def predict(text, speaker):
 
     # text = getNews ()
     # inputs = processor(text=text, return_tensors="pt")
-    inputs = processor(text=getNews(text), 
-                       return_tensors="pt")
 
+    if Interest == "":
+        inputs = processor(text=getNews(Interest), 
+                           return_tensors="pt")
+    else
+        inputs = processor(text=getNews(text), 
+                           return_tensors="pt")
+
+        
     # limit input length
     input_ids = inputs["input_ids"]
     input_ids = input_ids[..., :model.config.max_text_positions]
 
+    
     if speaker == "Surprise Me!":
         # load one of the provided speaker embeddings at random
         idx = np.random.randint(len(speaker_embeddings))
@@ -93,14 +100,15 @@ examples = [
 gr.Interface(
     fn=predict,
     inputs=[
-        gr.Text(label="Input Text"),
-        gr.Radio(label="Speaker", choices=[
-            "BDL (male)",
-            "CLB (female)",
-            "SLT (female)",
+        gr.Text(label="keyword"),
+        gr.Radio(label="Interest", choices=[
+            "Minecraft",
+            "Technology",
+            "Politics",
+            "KPop",
             "Surprise Me!"
         ],
-        value="BDL (male)"),
+        value="KPop"),
     ],
     outputs=[
         gr.Audio(label="Generated Speech", type="numpy"),