maybe timeout for gpu

app.py

@@ -3,16 +3,19 @@
 from __future__ import annotations
 
 import os
+import datetime
 
 import gradio as gr
 import spaces
 
 
-@spaces.GPU
+@spaces.GPU(duration=60 * 10)
 def run_on_gpu(input_shape, text, seed, guidance_scale):
+    print('Starting inference at {}'.format(datetime.datetime.now()))
     from model import Model
     model = Model()
     res_generator = model.run(shape_path=input_shape, text=text, seed=seed, guidance_scale=guidance_scale)
+    print('Finished inference at {}'.format(datetime.datetime.now()))
     return list(res_generator)