Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,7 @@
|
|
1 |
import gradio as gr
|
2 |
from transformers import MarianMTModel, MarianTokenizer
|
3 |
import torch
|
|
|
4 |
# Load the model and tokenizer from the Hub
|
5 |
model_name = "Dddixyy/latin-italian-translatorV2"
|
6 |
tokenizer = MarianTokenizer.from_pretrained(model_name)
|
@@ -8,6 +9,9 @@ model = MarianMTModel.from_pretrained(model_name)
|
|
8 |
|
9 |
# Translation function
|
10 |
def translate_latin_to_italian(latin_text):
|
|
|
|
|
|
|
11 |
inputs = tokenizer(latin_text, return_tensors="pt", padding=True, truncation=True)
|
12 |
with torch.no_grad():
|
13 |
generated_ids = model.generate(inputs["input_ids"])
|
|
|
1 |
import gradio as gr
|
2 |
from transformers import MarianMTModel, MarianTokenizer
|
3 |
import torch
|
4 |
+
|
5 |
# Load the model and tokenizer from the Hub
|
6 |
model_name = "Dddixyy/latin-italian-translatorV2"
|
7 |
tokenizer = MarianTokenizer.from_pretrained(model_name)
|
|
|
9 |
|
10 |
# Translation function
|
11 |
def translate_latin_to_italian(latin_text):
|
12 |
+
# Make the first letter lowercase if the input is not empty
|
13 |
+
if latin_text:
|
14 |
+
latin_text = latin_text[0].lower() + latin_text[1:]
|
15 |
inputs = tokenizer(latin_text, return_tensors="pt", padding=True, truncation=True)
|
16 |
with torch.no_grad():
|
17 |
generated_ids = model.generate(inputs["input_ids"])
|