Spaces:
Sleeping
Sleeping
un-index
commited on
Commit
·
0e7fd51
1
Parent(s):
22b8e06
app.py
CHANGED
|
@@ -10,7 +10,9 @@ import gradio as gr
|
|
| 10 |
ex=None
|
| 11 |
try:
|
| 12 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 13 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
|
|
|
|
|
|
| 14 |
# tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-j-6B")
|
| 15 |
# model = AutoModelForCausalLM.from_pretrained("EleutherAI/gpt-j-6B")
|
| 16 |
except Exception as e:
|
|
@@ -20,7 +22,7 @@ def f(text):
|
|
| 20 |
try:
|
| 21 |
if ex:
|
| 22 |
raise Exception("err from transformers import: \n"+str(ex))
|
| 23 |
-
model = AutoModelForCausalLM.from_pretrained("
|
| 24 |
default_do_sample = False #True (default is True)
|
| 25 |
input_ids = tokenizer(text, return_tensors="pt").input_ids
|
| 26 |
gen_tokens = model.generate(input_ids, do_sample=default_do_sample, temperature=0.9, max_length=30)
|
|
|
|
| 10 |
ex=None
|
| 11 |
try:
|
| 12 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 13 |
+
tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-j-6B")
|
| 14 |
+
|
| 15 |
+
# "EluttherAI" on this line and for the next occurence only
|
| 16 |
# tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-j-6B")
|
| 17 |
# model = AutoModelForCausalLM.from_pretrained("EleutherAI/gpt-j-6B")
|
| 18 |
except Exception as e:
|
|
|
|
| 22 |
try:
|
| 23 |
if ex:
|
| 24 |
raise Exception("err from transformers import: \n"+str(ex))
|
| 25 |
+
model = AutoModelForCausalLM.from_pretrained("EleutherAI/gpt-j-6B")
|
| 26 |
default_do_sample = False #True (default is True)
|
| 27 |
input_ids = tokenizer(text, return_tensors="pt").input_ids
|
| 28 |
gen_tokens = model.generate(input_ids, do_sample=default_do_sample, temperature=0.9, max_length=30)
|