Lora
commited on
Commit
·
8b1a6a7
1
Parent(s):
9da80c3
try add auth
Browse files
app.py
CHANGED
|
@@ -8,13 +8,12 @@ def visualize_word(word, count=10, remove_space=False):
|
|
| 8 |
|
| 9 |
if not remove_space:
|
| 10 |
word = ' ' + word
|
| 11 |
-
print(f"Looking up word
|
| 12 |
|
| 13 |
# seems very dumb to have to load the tokenizer every time, but I don't know how to pass a non-interface element into the function in gradio
|
| 14 |
tokenizer = transformers.AutoTokenizer.from_pretrained('gpt2')
|
| 15 |
vecs = torch.load("senses/all_vecs_mtx.pt")
|
| 16 |
lm_head = torch.load("senses/lm_head.pt")
|
| 17 |
-
print("lm_head.shape = ", lm_head.shape)
|
| 18 |
|
| 19 |
token_ids = tokenizer(word)['input_ids']
|
| 20 |
tokens = [tokenizer.decode(token_id) for token_id in token_ids]
|
|
@@ -87,5 +86,5 @@ with gr.Blocks() as demo:
|
|
| 87 |
outputs= [pos_outputs, neg_outputs, token_breakdown],
|
| 88 |
)
|
| 89 |
|
| 90 |
-
demo.launch(
|
| 91 |
|
|
|
|
| 8 |
|
| 9 |
if not remove_space:
|
| 10 |
word = ' ' + word
|
| 11 |
+
print(f"Looking up word '{word}'")
|
| 12 |
|
| 13 |
# seems very dumb to have to load the tokenizer every time, but I don't know how to pass a non-interface element into the function in gradio
|
| 14 |
tokenizer = transformers.AutoTokenizer.from_pretrained('gpt2')
|
| 15 |
vecs = torch.load("senses/all_vecs_mtx.pt")
|
| 16 |
lm_head = torch.load("senses/lm_head.pt")
|
|
|
|
| 17 |
|
| 18 |
token_ids = tokenizer(word)['input_ids']
|
| 19 |
tokens = [tokenizer.decode(token_id) for token_id in token_ids]
|
|
|
|
| 86 |
outputs= [pos_outputs, neg_outputs, token_breakdown],
|
| 87 |
)
|
| 88 |
|
| 89 |
+
demo.launch(auth=("admin", "pass1234"))
|
| 90 |
|