이 풀 리퀘스트를 사용하면 모든 오디오 세대가 서로 다를 수 있습니다./This PR makes each generation different

각 세대에서 시드를 무작위로 지정합니다.
Merge를 클릭하여 이 기능을 추가합니다.

It randomizes the seed at each generation.
Click on _Merge_ to add this feature.

app.py

@@ -1,3 +1,4 @@
+import random
 import torch
 import torchaudio
 from einops import rearrange
@@ -30,6 +31,11 @@ def generate_audio(prompt, seconds_total=30, steps=100, cfg_scale=7):
     translated_prompt = translator(prompt, max_length=512)[0]['translation_text']
     print(f"Translated Prompt: {translated_prompt}")
     
+    seed = random.randint(0, 2**63 - 1)
+    random.seed(seed)
+    torch.manual_seed(seed)
+    print(f"Using seed: {seed}")
+
     device = "cuda" if torch.cuda.is_available() else "cpu"
     print(f"Using device: {device}")