Spaces:
Running
on
T4
Running
on
T4
Update app.py
Browse files
app.py
CHANGED
|
@@ -113,8 +113,8 @@ STYLE = """
|
|
| 113 |
from transformers import GPT2Tokenizer, AutoModelForCausalLM
|
| 114 |
import numpy as np
|
| 115 |
|
| 116 |
-
tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
|
| 117 |
-
model = AutoModelForCausalLM.from_pretrained("gpt2")
|
| 118 |
tokenizer.pad_token_id = tokenizer.eos_token_id
|
| 119 |
pritn("Loading finished.")
|
| 120 |
def generate_html(token, node):
|
|
|
|
| 113 |
from transformers import GPT2Tokenizer, AutoModelForCausalLM
|
| 114 |
import numpy as np
|
| 115 |
|
| 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 |
pritn("Loading finished.")
|
| 120 |
def generate_html(token, node):
|