Fix pipeline_tag to text-generation
Browse files
README.md
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
---
|
| 2 |
-
pipeline_tag:
|
| 3 |
library_name: transformers
|
| 4 |
tags:
|
| 5 |
- t5
|
|
@@ -23,20 +23,6 @@ result = pipe("translate to SQL: list all users older than 18 | schema: users(id
|
|
| 23 |
print(result[0]["generated_text"])
|
| 24 |
```
|
| 25 |
|
| 26 |
-
## Inference API
|
| 27 |
-
|
| 28 |
-
```python
|
| 29 |
-
import requests
|
| 30 |
-
|
| 31 |
-
API_URL = "https://router.huggingface.co/hf-inference/models/RealMati/t2sql_v6_structured"
|
| 32 |
-
headers = {"Authorization": "Bearer hf_YOUR_TOKEN"}
|
| 33 |
-
|
| 34 |
-
response = requests.post(API_URL, headers=headers, json={
|
| 35 |
-
"inputs": "translate to SQL: How many users are there? | schema: users(id, name, age)"
|
| 36 |
-
})
|
| 37 |
-
print(response.json())
|
| 38 |
-
```
|
| 39 |
-
|
| 40 |
## Training
|
| 41 |
|
| 42 |
- **Base model:** T5-base
|
|
|
|
| 1 |
---
|
| 2 |
+
pipeline_tag: text-generation
|
| 3 |
library_name: transformers
|
| 4 |
tags:
|
| 5 |
- t5
|
|
|
|
| 23 |
print(result[0]["generated_text"])
|
| 24 |
```
|
| 25 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 26 |
## Training
|
| 27 |
|
| 28 |
- **Base model:** T5-base
|