urwebsiteaz-ux commited on
Commit
adadf29
·
1 Parent(s): c05afe6

Fix adapter repo name

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ from peft import PeftModel
9
 
10
  BASE_MODEL = "openai/whisper-small"
11
  ADAPTER_REPO = os.environ.get(
12
- "ADAPTER_REPO", "YOURNAME/autolyrics-whisper-small-lora")
13
 
14
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
15
  DTYPE = torch.float16 if DEVICE == "cuda" else torch.float32
 
9
 
10
  BASE_MODEL = "openai/whisper-small"
11
  ADAPTER_REPO = os.environ.get(
12
+ "ADAPTER_REPO", "Petercoder/autolyrics-whisper-small-lora")
13
 
14
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
15
  DTYPE = torch.float16 if DEVICE == "cuda" else torch.float32