Warn the user when the calculations are aborted.

app.py

@@ -97,16 +97,22 @@ def compute_indices(
 
 # LOGITS #######################################################################
 
-@spaces.GPU(duration=60)
+@spaces.GPU(duration=30)
 def compute_logits(
     indices_arr: object,
 ) -> object:
-    # load the model and tokenizer inside the GPU wrapper
+    __logits = None
+    # load the model inside the GPU wrapper (not before)
     __model = fetch_model()
-    # fill all the arguments that cannot be pickled
-    return _app.update_logits_state(
-        indices_arr=indices_arr,
-        model_obj=__model)
+    # the allocation might expire before the calculations are finished
+    try:
+        __logits = _app.update_logits_state(
+            indices_arr=indices_arr,
+            model_obj=__model)
+    except:
+        gradio.Warning(title='Warning', message='Calculations aborted because the GPU allocation expired.', duration=4)
+    # tensor or None
+    return __logits
 
 # MAIN #########################################################################