suarkadipa commited on
Commit
1d57f1e
1 Parent(s): b06a6b3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -5
README.md CHANGED
@@ -28,10 +28,9 @@ More information needed
28
 
29
  # How to run in Google Colab
30
 
 
31
  from transformers import AutoTokenizer, AutoModelForCausalLM
32
-
33
  tokenizer_fromhub = AutoTokenizer.from_pretrained("suarkadipa/GPT-2-finetuned-papers")
34
-
35
  model_fromhub = AutoModelForCausalLM.from_pretrained("suarkadipa/GPT-2-finetuned-papers", from_tf=True)
36
 
37
  text_generator = pipeline(
@@ -43,12 +42,10 @@ text_generator = pipeline(
43
  )
44
 
45
  // change with your text
46
-
47
  test_sentence = "the role of recommender systems"
48
-
49
  res=text_generator(test_sentence)[0]["generated_text"].replace("\n", " ")
50
-
51
  res
 
52
 
53
  ## Training and evaluation data
54
 
 
28
 
29
  # How to run in Google Colab
30
 
31
+ ```python
32
  from transformers import AutoTokenizer, AutoModelForCausalLM
 
33
  tokenizer_fromhub = AutoTokenizer.from_pretrained("suarkadipa/GPT-2-finetuned-papers")
 
34
  model_fromhub = AutoModelForCausalLM.from_pretrained("suarkadipa/GPT-2-finetuned-papers", from_tf=True)
35
 
36
  text_generator = pipeline(
 
42
  )
43
 
44
  // change with your text
 
45
  test_sentence = "the role of recommender systems"
 
46
  res=text_generator(test_sentence)[0]["generated_text"].replace("\n", " ")
 
47
  res
48
+ ```
49
 
50
  ## Training and evaluation data
51