Trying zerogpu

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from transformers import NllbTokenizer, M2M100ForConditionalGeneration
 import torch
+import spaces
 from huggingface_hub import login, snapshot_download
 import os
 
@@ -41,6 +42,7 @@ language_tokens = {
 
 supported_languages = list(language_tokens.keys())
 
+@spaces.gpu
 def translate(text, source_language, target_language):
     if source_language not in supported_languages:
         raise ValueError(f"Source language '{source_language}' not supported. Supported: {supported_languages}")