Omlesna commited on
Commit
3e21b80
·
1 Parent(s): ffa5609

fix remote bug

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -102,7 +102,10 @@ def get_hf_token(profile: gr.OAuthProfile | None):
102
  print("Using token from environment.")
103
  if not token:
104
  try:
105
- from huggingface_hub import HfFolder
 
 
 
106
  token = HfFolder.get_token()
107
  if token:
108
  print("Using token from local HF cache (huggingface-cli login).")
 
102
  print("Using token from environment.")
103
  if not token:
104
  try:
105
+ try:
106
+ from huggingface_hub import HfFolder # older versions
107
+ except Exception:
108
+ from huggingface_hub.hf_api import HfFolder # fallback for space runtime
109
  token = HfFolder.get_token()
110
  if token:
111
  print("Using token from local HF cache (huggingface-cli login).")