Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import spaces
 from transformers import pipeline
 from typing import Union, List
 
@@ -8,8 +9,8 @@ class BambaraTranslator:
             "translation",
             model=model_name,
             max_length=512,
-            truncation=True ,
-            device=-1  # Force CPU
+            truncation=True,
+            device=0 if spaces.config.Hardware.current == "cuda" else -1  
         )
         self.flores_codes = {
             "French": "fra_Latn",
@@ -17,6 +18,7 @@ class BambaraTranslator:
             "Bambara": "bam_Latn"
         }
 
+    @spaces.GPU
     def translate(self, text: Union[str, List[str]], src_lang: str, tgt_lang: str) -> Union[str, List[str]]:
         source_lang = self.flores_codes[src_lang]
         target_lang = self.flores_codes[tgt_lang]
@@ -36,6 +38,7 @@ examples = [
     "The sun rises every morning to bring light to the world."
 ]
 
+@spaces.GPU
 def translate_text(src_lang, tgt_lang, text):
     if not text.strip():
         return "Please enter text to translate."
@@ -75,7 +78,7 @@ with gr.Blocks(title="Bambara Translator") as demo:
     
     gr.Markdown(
         """
-        By [sudoping01](https://huggingface.co/sudoping01), from [sudoping01/nllb-bambara-v2](https://huggingface.co/sudoping01/nllb-bambara-v2). Fine-tuned on Meta’s NLLB, CC BY-NC 4.0, non-commercial.
+        By [sudoping01](https://huggingface.co/sudoping01), from [sudoping01/nllb-bambara-v2](https://huggingface.co/sudoping01/nllb-bambara-v2). Fine-tuned on Meta's NLLB, CC BY-NC 4.0, non-commercial.
         """
     )