rverma0631 commited on
Commit
1034cbe
·
verified ·
1 Parent(s): 75c4f2e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -1
app.py CHANGED
@@ -2,6 +2,17 @@ import gradio as gr
2
  import torch
3
  import torchaudio
4
  from transformers import AutoModel
 
 
 
 
 
 
 
 
 
 
 
5
 
6
  # Initialize device
7
  device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -11,7 +22,8 @@ print(f"Using device: {device}")
11
  print("Loading IndicConformer model...")
12
  indic_asr_model = AutoModel.from_pretrained(
13
  "ai4bharat/indic-conformer-600m-multilingual",
14
- trust_remote_code=True
 
15
  )
16
  if device == "cuda":
17
  indic_asr_model = indic_asr_model.to(device)
 
2
  import torch
3
  import torchaudio
4
  from transformers import AutoModel
5
+ from huggingface_hub import login
6
+ import os
7
+
8
+ # Authenticate with Hugging Face
9
+ # The token will be automatically available in HF Spaces as an environment variable
10
+ hf_token = os.getenv("HF_TOKEN")
11
+ if hf_token:
12
+ login(token=hf_token)
13
+ print("✅ Authenticated with Hugging Face")
14
+ else:
15
+ print("⚠️ HF_TOKEN not found. Make sure to add it in Space settings.")
16
 
17
  # Initialize device
18
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
22
  print("Loading IndicConformer model...")
23
  indic_asr_model = AutoModel.from_pretrained(
24
  "ai4bharat/indic-conformer-600m-multilingual",
25
+ trust_remote_code=True,
26
+ token=hf_token # Pass token explicitly
27
  )
28
  if device == "cuda":
29
  indic_asr_model = indic_asr_model.to(device)