Spaces:
Running
Running
app.py
CHANGED
@@ -11,7 +11,7 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
|
11 |
|
12 |
# API URLs and headers
|
13 |
AUDIO_API_URL = "https://api-inference.huggingface.co/models/MIT/ast-finetuned-audioset-10-10-0.4593"
|
14 |
-
JANUS_API_URL = "https://api-inference.huggingface.co/models/deepseek-ai/Janus-1.3B"
|
15 |
headers = {"Authorization": f"Bearer {os.environ.get('HF_TOKEN')}"}
|
16 |
|
17 |
def format_error(message):
|
@@ -53,13 +53,20 @@ def generate_lyrics_with_retry(prompt, max_retries=5, initial_wait=2):
|
|
53 |
"max_new_tokens": 200,
|
54 |
"temperature": 0.7,
|
55 |
"top_p": 0.9,
|
|
|
56 |
"return_full_text": False
|
57 |
}
|
58 |
}
|
59 |
)
|
60 |
|
|
|
|
|
|
|
61 |
if response.status_code == 200:
|
62 |
-
|
|
|
|
|
|
|
63 |
elif response.status_code == 503:
|
64 |
print(f"Model loading, attempt {attempt + 1}/{max_retries}. Waiting {wait_time} seconds...")
|
65 |
time.sleep(wait_time)
|
|
|
11 |
|
12 |
# API URLs and headers
|
13 |
AUDIO_API_URL = "https://api-inference.huggingface.co/models/MIT/ast-finetuned-audioset-10-10-0.4593"
|
14 |
+
JANUS_API_URL = "https://api-inference.huggingface.co/models/deepseek-ai/Janus-1.3B/text-generation"
|
15 |
headers = {"Authorization": f"Bearer {os.environ.get('HF_TOKEN')}"}
|
16 |
|
17 |
def format_error(message):
|
|
|
53 |
"max_new_tokens": 200,
|
54 |
"temperature": 0.7,
|
55 |
"top_p": 0.9,
|
56 |
+
"do_sample": True,
|
57 |
"return_full_text": False
|
58 |
}
|
59 |
}
|
60 |
)
|
61 |
|
62 |
+
print(f"Response status: {response.status_code}")
|
63 |
+
print(f"Response content: {response.content.decode('utf-8', errors='ignore')}")
|
64 |
+
|
65 |
if response.status_code == 200:
|
66 |
+
result = response.json()
|
67 |
+
if isinstance(result, list) and len(result) > 0:
|
68 |
+
return result[0].get("generated_text", "Error: No text generated")
|
69 |
+
return "Error: Unexpected response format"
|
70 |
elif response.status_code == 503:
|
71 |
print(f"Model loading, attempt {attempt + 1}/{max_retries}. Waiting {wait_time} seconds...")
|
72 |
time.sleep(wait_time)
|