suarkadipa commited on
Commit
dfc664a
·
1 Parent(s): 723faf1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +19 -1
README.md CHANGED
@@ -26,7 +26,25 @@ More information needed
26
 
27
  ## Intended uses & limitations
28
 
29
- More information needed
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
30
 
31
  ## Training and evaluation data
32
 
 
26
 
27
  ## Intended uses & limitations
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
+ model_fromhub = AutoModelForCausalLM.from_pretrained("suarkadipa/GPT-2-finetuned-papers", from_tf=True)
35
+
36
+ text_generator = pipeline(
37
+ "text-generation",
38
+ model=model_fromhub,
39
+ tokenizer=tokenizer_fromhub,
40
+ framework="tf",
41
+ max_new_tokens=3000
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
  ## Training and evaluation data
50