import gradio as gr
import time
import spaces

@spaces.GPU
def test(x, y):
    time.sleep(3)
    return x

gr.ChatInterface(test).launch()