Spaces:
Runtime error
Runtime error
HatmanStack
commited on
Commit
·
8d7e10f
1
Parent(s):
ebe993d
added token
Browse files
app.py
CHANGED
|
@@ -6,10 +6,12 @@ import random
|
|
| 6 |
import numpy as np
|
| 7 |
import spaces
|
| 8 |
import json
|
|
|
|
| 9 |
|
|
|
|
| 10 |
cuda_device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
| 11 |
model_checkpoint = "Hatman/Flux-Prompt-Enhance"
|
| 12 |
-
enhancer_tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
|
| 13 |
enhancer_model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint).to(cuda_device)
|
| 14 |
|
| 15 |
model_path = "deepseek-ai/Janus-Pro-7B"
|
|
|
|
| 6 |
import numpy as np
|
| 7 |
import spaces
|
| 8 |
import json
|
| 9 |
+
import os
|
| 10 |
|
| 11 |
+
token = os.env.get('HF_TOKEN')
|
| 12 |
cuda_device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
| 13 |
model_checkpoint = "Hatman/Flux-Prompt-Enhance"
|
| 14 |
+
enhancer_tokenizer = AutoTokenizer.from_pretrained(model_checkpoint, token=token)
|
| 15 |
enhancer_model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint).to(cuda_device)
|
| 16 |
|
| 17 |
model_path = "deepseek-ai/Janus-Pro-7B"
|