multimodalart HF Staff commited on
Commit
33b909c
·
1 Parent(s): c4b4ec8

[Admin maintenance] Support new ZeroGPU hardware (#2)

Browse files

- [Admin maintenance] Support new ZeroGPU hardware (32a87bd1c492e8ce7b211427637ca760247d6998)

Files changed (1) hide show
  1. app.py +14 -0
app.py CHANGED
@@ -1,3 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import torch
2
  import gradio as gr
3
  from PIL import Image
 
1
+ # ZeroGPU/Blackwell maintenance: restore HfFolder (removed from newer huggingface_hub, still
2
+ # imported by older gradio/oauth.py). Must run before `import spaces` (which imports gradio).
3
+ import huggingface_hub
4
+ if not hasattr(huggingface_hub, "HfFolder"):
5
+ class HfFolder:
6
+ @staticmethod
7
+ def get_token(): return huggingface_hub.get_token()
8
+ @staticmethod
9
+ def save_token(token): huggingface_hub.login(token)
10
+ @staticmethod
11
+ def delete_token():
12
+ try: huggingface_hub.logout()
13
+ except Exception: pass
14
+ huggingface_hub.HfFolder = HfFolder
15
  import torch
16
  import gradio as gr
17
  from PIL import Image