Add zerogpu

app.py

@@ -1,6 +1,6 @@
 from functools import partial
 from pathlib import Path
-
+import spaces
 import gradio as gr
 from joeynmt.datasets import build_dataset
 from joeynmt.helpers import (
@@ -30,6 +30,7 @@ languages_scripts = {
 }
 
 
+@spaces.GPU
 def normalize(text, language_script):
     cfg_file = "./models/%s/config.yaml" % languages_scripts[language_script]
     ckpt = "./models/%s/best.ckpt" % languages_scripts[language_script]