Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -116,7 +116,7 @@ import numpy as np
|
|
116 |
tokenizer = GPT2Tokenizer.from_pretrained("openai-community/gpt2")
|
117 |
model = AutoModelForCausalLM.from_pretrained("openai-community/gpt2")
|
118 |
tokenizer.pad_token_id = tokenizer.eos_token_id
|
119 |
-
|
120 |
def generate_html(token, node):
|
121 |
"""Recursively generate HTML for the tree."""
|
122 |
|
|
|
116 |
tokenizer = GPT2Tokenizer.from_pretrained("openai-community/gpt2")
|
117 |
model = AutoModelForCausalLM.from_pretrained("openai-community/gpt2")
|
118 |
tokenizer.pad_token_id = tokenizer.eos_token_id
|
119 |
+
print("Loading finished.")
|
120 |
def generate_html(token, node):
|
121 |
"""Recursively generate HTML for the tree."""
|
122 |
|