suarkadipa
commited on
Commit
·
1d57f1e
1
Parent(s):
b06a6b3
Update README.md
Browse files
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 |
|