lanzhiwang commited on
Commit
8fe72f7
·
1 Parent(s): 872c8d7
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -1,12 +1,12 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
- # from transformers import AutoTokenizer, AutoModelForCausalLM
4
 
5
- # tokenizer = AutoTokenizer.from_pretrained(pretrained_model_name_or_path="/root/.cache/huggingface/hub/models")
6
- # model = AutoModelForCausalLM.from_pretrained(pretrained_model_name_or_path="/root/.cache/huggingface/hub/models")
7
- # generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
8
 
9
- generator = pipeline('text-generation', model='gpt2')
10
 
11
  def generate(text):
12
  result = generator(text, max_length=30, num_return_sequences=1)
 
1
  import gradio as gr
2
  from transformers import pipeline
3
+ from transformers import AutoTokenizer, AutoModelForCausalLM
4
 
5
+ tokenizer = AutoTokenizer.from_pretrained(pretrained_model_name_or_path="/root/.cache/huggingface/hub/models")
6
+ model = AutoModelForCausalLM.from_pretrained(pretrained_model_name_or_path="/root/.cache/huggingface/hub/models")
7
+ generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
8
 
9
+ # generator = pipeline('text-generation', model='gpt2')
10
 
11
  def generate(text):
12
  result = generator(text, max_length=30, num_return_sequences=1)