first commit

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from transformers import pipeline
 import torch
 import logging
+import spaces
 from typing import Literal, Tuple
 
 # Set up logging
@@ -40,6 +41,7 @@ TARGET_LANG_OPTIONS = {
 
 
 # Define the translation function with typing
+@spaces.GPU()
 def translate_text(text: str, source_lang: str, target_lang: str) -> str:
     """
     Translate the input text from the source language to the target language using the NLLB model.