update app.py

app.py

@@ -32,7 +32,7 @@ if torch.cuda.is_available():
 setup_eval_logging()
 OUTPUT_DIR = Path("./output/gradio")
 OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
-NUM_SAMPLE=5
+NUM_SAMPLE=8
 snapshot_download(repo_id="google/flan-t5-large")
 a=AutoModel.from_pretrained('bert-base-uncased')
 b=AutoModel.from_pretrained('roberta-base')
@@ -207,6 +207,8 @@ def generate_audio_gradio(
     scores = torch.cosine_similarity(text_embed.expand(audio_embed.shape[0], -1),
                                 audio_embed,
                                 dim=-1)
+    log.info(scores)
+    log.info(torch.argmax(scores).item())
     audio=audios[torch.argmax(scores).item()].float().cpu()
     safe_prompt = (
         "".join(c for c in prompt if c.isalnum() or c in (" ", "_"))