Update app.py

app.py

@@ -1,21 +1,21 @@
-import gradio as gr
-from load_model import pipe
-from utils import Translation
-
-
-
-def reply(message, history):
-    txt = Translation(message, "en")
-    if txt.original == "en":
-        response = pipe(message)
-        return response[0]["generated_text"]
-    else:
-        translation = txt.translatef()
-        response = pipe(translation)
-        t = Translation(response[0]["generated_text"], txt.original)
-        res = t.translatef()
-        return res
-
-
-demo = gr.ChatInterface(fn=reply, title="Multilingual-Bloom Bot")
+import gradio as gr
+from load_model import pipe
+from utils import Translation
+
+
+@spaces.GPU
+def reply(message, history):
+    txt = Translation(message, "en")
+    if txt.original == "en":
+        response = pipe(message)
+        return response[0]["generated_text"]
+    else:
+        translation = txt.translatef()
+        response = pipe(translation)
+        t = Translation(response[0]["generated_text"], txt.original)
+        res = t.translatef()
+        return res
+
+
+demo = gr.ChatInterface(fn=reply, title="Multilingual-Bloom Bot")
 demo.launch()