Update app.py

app.py

@@ -4,7 +4,7 @@ from transformers import T5Tokenizer, T5ForConditionalGeneration, AutoTokenizer,
 
 langs = {"English": "en", "Romanian": "ro", "German": "de", "French": "fr", "Spanish": "es", "Italian": "it"}
 options = list(langs.keys())
-models = ["Helsinki-NLP", "t5-base", "t5-small", "t5-large", "madlad400-10b-mt"]
+models = ["Helsinki-NLP", "t5-base", "t5-small", "t5-large", "google/madlad400-10b-mt"]
 
 @spaces.GPU
 def translate_text(input_text, sselected_language, tselected_language, model_name):
@@ -25,7 +25,7 @@ def translate_text(input_text, sselected_language, tselected_language, model_nam
                 
     else:
         tokenizer = T5Tokenizer.from_pretrained(model_name)
-        model = T5ForConditionalGeneration.from_pretrained(model_name)
+        model = T5ForConditionalGeneration.from_pretrained(model_name, device_map="auto")
 
     if model_name.startswith("Helsinki-NLP"):
         prompt = input_text