NoufSaleh46 commited on
Commit
82f2474
·
verified ·
1 Parent(s): 37d987e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -5,6 +5,7 @@ from transformers import pipeline
5
  gen_model = pipeline("text2text-generation", model="google/flan-t5-large")
6
  translator_en_ar = pipeline("translation", model="Helsinki-NLP/opus-mt-en-ar") # English to Arabic
7
  translator_ar_en = pipeline("translation", model="Helsinki-NLP/opus-mt-ar-en") # Arabic to English
 
8
 
9
  def get_plant_info(plant_name, language):
10
  if language == "English":
 
5
  gen_model = pipeline("text2text-generation", model="google/flan-t5-large")
6
  translator_en_ar = pipeline("translation", model="Helsinki-NLP/opus-mt-en-ar") # English to Arabic
7
  translator_ar_en = pipeline("translation", model="Helsinki-NLP/opus-mt-ar-en") # Arabic to English
8
+ tokenizer = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-ar")
9
 
10
  def get_plant_info(plant_name, language):
11
  if language == "English":