SreyanG-NVIDIA commited on
Commit
4b399db
·
1 Parent(s): add7e19

Fix GitHub badge

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -99,7 +99,7 @@ def _log_cuda_runtime(stage):
99
  print(f"[{stage}] failed to inspect CUDA devices: {e}", flush=True)
100
 
101
 
102
- REPO_URL = "https://github.com/afnext-umd-nvidia/afnext-umd-nvidia.github.io"
103
  MODEL_ID = "nvidia/audio-flamingo-next-hf"
104
  HERO_IMAGE_URL = "https://afnext-umd-nvidia.github.io/logo.webp"
105
  HERO_TITLE = "Audio Flamingo Next: Next-Generation Open Audio-Language Models for Speech, Sound, and Music"
@@ -121,7 +121,7 @@ HERO_BADGES = f"""
121
  <div style="display: flex; justify-content: center; margin-top: 6px; align-items: center;">
122
  <div style="display: flex; justify-content: center; flex-wrap: wrap; gap: 8px;">
123
  <a href="https://afnext-umd-nvidia.github.io/"><img src="https://img.shields.io/badge/Project%20Page-AF--Next-0F766E" alt="Project Page"></a>
124
- <a href="{REPO_URL}"><img src='https://img.shields.io/badge/GitHub-AF--Next-0E7490' alt="GitHub"></a>
125
  <a href="https://huggingface.co/nvidia/audio-flamingo-next-hf">
126
  <img src="https://img.shields.io/badge/🤗-Model%20Checkpoint-ED5A22.svg" alt="Model Checkpoint">
127
  </a>
 
99
  print(f"[{stage}] failed to inspect CUDA devices: {e}", flush=True)
100
 
101
 
102
+ CODE_URL = "https://github.com/NVIDIA/audio-flamingo"
103
  MODEL_ID = "nvidia/audio-flamingo-next-hf"
104
  HERO_IMAGE_URL = "https://afnext-umd-nvidia.github.io/logo.webp"
105
  HERO_TITLE = "Audio Flamingo Next: Next-Generation Open Audio-Language Models for Speech, Sound, and Music"
 
121
  <div style="display: flex; justify-content: center; margin-top: 6px; align-items: center;">
122
  <div style="display: flex; justify-content: center; flex-wrap: wrap; gap: 8px;">
123
  <a href="https://afnext-umd-nvidia.github.io/"><img src="https://img.shields.io/badge/Project%20Page-AF--Next-0F766E" alt="Project Page"></a>
124
+ <a href="{CODE_URL}"><img src='https://img.shields.io/badge/GitHub-AF--Next-0E7490' alt="GitHub"></a>
125
  <a href="https://huggingface.co/nvidia/audio-flamingo-next-hf">
126
  <img src="https://img.shields.io/badge/🤗-Model%20Checkpoint-ED5A22.svg" alt="Model Checkpoint">
127
  </a>