MrFrijo commited on
Commit
86fc274
·
verified ·
1 Parent(s): e0ee38f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -1,4 +1,11 @@
1
  import gradio as gr
 
 
 
 
 
 
 
2
 
3
  def translate_text(text, src_lang, target_lang):
4
  # Préparer les entrées pour le modèle
 
1
  import gradio as gr
2
+ from transformers import MarianMTModel, MarianTokenizer
3
+
4
+ # Charger le modèle et le tokenizer depuis Hugging Face
5
+ model_name = "MrFrijo/LiAPI/" # Nom du modèle sur Hugging Face
6
+
7
+ model = MarianMTModel.from_pretrained(model_name)
8
+ tokenizer = MarianTokenizer.from_pretrained(model_name)
9
 
10
  def translate_text(text, src_lang, target_lang):
11
  # Préparer les entrées pour le modèle