gagan3012 commited on
Commit
bd89baa
1 Parent(s): 83d33d6

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +33 -0
README.md ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # keytotext
2
+
3
+ Idea is to build a model which will take keywords as inputs and generate sentences as outputs.
4
+
5
+ ### Model:
6
+
7
+ Two Models have been built:
8
+
9
+ - Using T5-base size = 850 MB can be found here: https://huggingface.co/gagan3012/keytotext
10
+ - Using T5-small size = 230 MB can be found here: https://huggingface.co/gagan3012/keytotext-small
11
+
12
+ #### Usage:
13
+
14
+ ```python
15
+ from transformers import AutoTokenizer, AutoModelWithLMHead
16
+
17
+ tokenizer = AutoTokenizer.from_pretrained("gagan3012/keytotext-small")
18
+
19
+ model = AutoModelWithLMHead.from_pretrained("gagan3012/keytotext-small")
20
+ ```
21
+
22
+ ### Demo:
23
+
24
+ [![Streamlit App](https://static.streamlit.io/badges/streamlit_badge_black_white.svg)](https://share.streamlit.io/gagan3012/keytotext/app.py)
25
+
26
+ https://share.streamlit.io/gagan3012/keytotext/app.py
27
+
28
+ ![image](https://user-images.githubusercontent.com/49101362/110660053-3b20fe80-81d4-11eb-9275-ba402134e8d9.png)
29
+
30
+
31
+ ### Example:
32
+
33
+ ['India', 'Wedding'] -> We are celebrating today in New Delhi with three wedding anniversary parties.