Jesus Carrasco commited on
Commit
a071f32
1 Parent(s): 3d23272

Delete main.py

Browse files
Files changed (1) hide show
  1. main.py +0 -38
main.py DELETED
@@ -1,38 +0,0 @@
1
- import gradio as gr
2
- from transformers import MarianMTModel, MarianTokenizer
3
-
4
- # Define the translator function
5
- def translate_text(text, target_language):
6
- model_name = {
7
- "Spanish": "Helsinki-NLP/opus-mt-en-es",
8
- "French": "Helsinki-NLP/opus-mt-en-fr",
9
- "Italian": "Helsinki-NLP/opus-mt-en-it",
10
- "Ukrainian": "Helsinki-NLP/opus-mt-en-uk",
11
- }
12
-
13
- tokenizer = MarianTokenizer.from_pretrained(model_name[target_language])
14
- model = MarianMTModel.from_pretrained(model_name[target_language])
15
-
16
- # Tokenize the input text
17
- input_tokens = tokenizer(text, return_tensors="pt")
18
-
19
- # Generate translated tokens
20
- translated_tokens = model.generate(**input_tokens)
21
-
22
- # Decode the translated tokens
23
- translated_text = tokenizer.decode(translated_tokens[0], skip_special_tokens=True)
24
-
25
- return translated_text
26
-
27
- # Define the Gradio interface
28
- language_options = ["Spanish", "French", "Italian", "Ukrainian"]
29
- inputs = [
30
- gr.inputs.Textbox(lines=5, label="Enter text to translate:"),
31
- gr.inputs.Dropdown(choices=language_options, label="Select target language:"),
32
- ]
33
- outputs = gr.outputs.Textbox(label="Translated text:")
34
-
35
- iface = gr.Interface(fn=translate_text, inputs=inputs, outputs=outputs, title="Language Translator")
36
-
37
- # Launch the Hugging Face Spaces app
38
- iface.launch()