Commit ·
dfc664a
1
Parent(s): 723faf1
Update README.md
Browse files
README.md
CHANGED
|
@@ -26,7 +26,25 @@ More information needed
|
|
| 26 |
|
| 27 |
## Intended uses & limitations
|
| 28 |
|
| 29 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
|