softwareweaver commited on
Commit
7d021b9
·
verified ·
1 Parent(s): a7fb3c7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -64,6 +64,9 @@ def infer(
64
  if hf_token is None:
65
  raise RuntimeError("HF_TOKEN_SPACES is not available inside GPU worker.")
66
 
 
 
 
67
  # reload pipeline if user picked a different model
68
  if model_name != current_model_id:
69
  pipe = DiffusionPipeline.from_pretrained(
 
64
  if hf_token is None:
65
  raise RuntimeError("HF_TOKEN_SPACES is not available inside GPU worker.")
66
 
67
+ api = HfApi()
68
+ print("WHOAMI:", api.whoami(token=hf_token))
69
+
70
  # reload pipeline if user picked a different model
71
  if model_name != current_model_id:
72
  pipe = DiffusionPipeline.from_pretrained(