Mr-Vicky-01 commited on
Commit
34d8db2
·
verified ·
1 Parent(s): 8c12699

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +39 -1
README.md CHANGED
@@ -5,4 +5,42 @@ datasets: Hemanth-thunder/en_ta
5
  language:
6
  - ta
7
  - en
8
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5
  language:
6
  - ta
7
  - en
8
+ widget:
9
+ - text: >-
10
+ Actor Vijay is competing an 2026 election.
11
+ - text: >-
12
+ you need to study well for exams
13
+ ---
14
+
15
+ ## Model Details
16
+ - **Model Name**: English-Tamil-Translator
17
+ - **Model Type**: Deep Learning Model
18
+ - **Language**: Python
19
+ - **Task**: Language Translation
20
+
21
+ ## How to Use
22
+ 1. **Install Gemma Python Package**:
23
+ ```bash
24
+ pip install -q -U transformers==4.38.0
25
+ ```
26
+
27
+ ## Inference
28
+ 1. **How to use the model in our notebook**:
29
+ ```python
30
+ # Load model directly
31
+ import torch
32
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
33
+
34
+ checkpoint = "Mr-Vicky-01/English-Tamil-Translator"
35
+ tokenizer = AutoTokenizer.from_pretrained(checkpoint)
36
+ model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint)
37
+
38
+ def language_translator(text):
39
+ tokenized = tokenizer([text], return_tensors='pt')
40
+ out = model.generate(**tokenized, max_length=128)
41
+ return tokenizer.decode(out[0],skip_special_tokens=True)
42
+
43
+ text_to_translate = "i have to play football now!"
44
+ output = language_translator(text_to_translate)
45
+ print(output)
46
+ ```