Update app.py

app.py

@@ -1,14 +1,16 @@
+import spaces
 from transformers import AutoModelForSeq2SeqLM,AutoTokenizer, pipeline
 import gradio as grad
 mdl_name = "oddadmix/masrawy-english-arabic-translator"
 # mdl = AutoModelForSeq2SeqLM.from_pretrained(mdl_name)
 # my_tkn = AutoTokenizer.from_pretrained(mdl_name)
+pipe = pipeline("translation", model=mdl_name, device = 'cuda')
 
-pipe = pipeline("translation", model=mdl_name)
-
+@spaces.GPU
 def translate(text):
     # inputs = my_tkn(text, return_tensors="pt")
     # trans_output = mdl.generate(**inputs)
+    
     response = pipe(text)[0]['translation_text']
 
     #response = opus_translator(text)