badrex commited on
Commit
5437aa4
·
verified ·
1 Parent(s): e71fd1d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -4,6 +4,12 @@ import gradio as gr
4
  import spaces
5
  import torch
6
  from transformers import AutoProcessor, AutoModelForCTC
 
 
 
 
 
 
7
 
8
  device = "cuda" if torch.cuda.is_available() else "cpu"
9
  print(f"Using device: {device}")
 
4
  import spaces
5
  import torch
6
  from transformers import AutoProcessor, AutoModelForCTC
7
+ from huggingface_hub import login
8
+
9
+
10
+ HF_TOKEN = os.environ.get("HF_API_KEY")
11
+ if HF_TOKEN:
12
+ login(token=HF_TOKEN)
13
 
14
  device = "cuda" if torch.cuda.is_available() else "cpu"
15
  print(f"Using device: {device}")