un-index commited on
Commit
0e7fd51
·
1 Parent(s): 22b8e06
Files changed (1) hide show
  1. app.py +4 -2
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("huggingface/EleutherAI/gpt-j-6B")
 
 
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("huggingface/EleutherAI/gpt-j-6B")
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)