Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import torch
 import os
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+import spaces
 
 # Load Hugging Face token from the environment variable
 HF_TOKEN = os.getenv("HF_TOKEN")
@@ -10,7 +11,8 @@ if HF_TOKEN is None:
 
 # Check for GPU support and configure appropriately
 device = "cuda" if torch.cuda.is_available() else "cpu"
-print(f"Device being used: {device}")
+zero = torch.Tensor([0]).to(device)
+print(f"Device being used: {zero.device}")
 
 # Model configurations
 MSA_TO_SYRIAN_MODEL = "Omartificial-Intelligence-Space/Shami-MT"
@@ -27,6 +29,7 @@ syrian_to_msa_model = AutoModelForSeq2SeqLM.from_pretrained(SYRIAN_TO_MSA_MODEL)
 
 print("Models loaded successfully!")
 
+@spaces.GPU(duration=120)
 def translate_msa_to_syrian(text):
     """Translate from Modern Standard Arabic to Syrian dialect"""
     if not text.strip():
@@ -40,6 +43,7 @@ def translate_msa_to_syrian(text):
     except Exception as e:
         return f"Translation error: {str(e)}"
 
+@spaces.GPU(duration=120)
 def translate_syrian_to_msa(text):
     """Translate from Syrian dialect to Modern Standard Arabic"""
     if not text.strip():