Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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(
|