Update app.py

app.py

@@ -2,6 +2,7 @@ import os
 import torch
 import gradio as gr
 import time
+import spaces
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 from flores200_codes import flores_codes
 
@@ -22,6 +23,7 @@ def load_models():
     return model_dict
 
 
+@spaces.GPU
 def translation(source, target, text):
     if len(model_dict) == 2:
         model_name = 'nllb-distilled-1.3B'
@@ -65,10 +67,10 @@ if __name__ == '__main__':
 
     title = "NLLB distilled 1.3B demo"
 
-    demo_status = "Demo is running on CPU"
+    demo_status = "Demo is running on A10G GPU"
     description = f"Details: https://github.com/facebookresearch/fairseq/tree/nllb. {demo_status}"
     examples = [
-    ['English', 'Korean', 'Hi. nice to meet you']
+    ['English', 'Russian', 'Hi. nice to meet you']
     ]
 
     gr.Interface(translation,