Update app.py

app.py

@@ -115,7 +115,7 @@ def format_input(src, tgt_lang, src_lang, domain):
 
     return base_input
 
-
+@spaces.GPU(duration=120)
 def translate_with_model(model_name, text, tgt_lang, src_lang, domain):
     model = MODELS[model_name]
     formatted_text = format_input(text, tgt_lang, src_lang, domain)
@@ -238,7 +238,7 @@ def translate_with_model(model_name, text, tgt_lang, src_lang, domain):
             "translation": generated_translation,
         }
 
-
+@spaces.GPU(duration=120)
 def translate_with_all_models(selected_models, text, tgt_lang, src_lang, domain):
     tgt_lang = LANG2CODE[tgt_lang]
     src_lang = None if src_lang == "Auto" else LANG2CODE.get(src_lang)