handle gpu quota

app.py

@@ -962,6 +962,7 @@ def run_fn(
     # print(kwargs)
     
     try:
+        # try to aquiare GPU, can fail if the user is out of GPU quota
     
         if old_school_ncut:
             return super_duper_long_run(model, images, **kwargs)
@@ -994,14 +995,18 @@ def run_fn(
         return quick_run(model, images, **kwargs)
     
     except gr.Error as e:
-        print(e)
-        gr.Warning(e.message)
-        if USE_HUGGINGFACE_ZEROGPU:
-            spaces.zero.gradio.HTMLError(e.message)
-        gr.Warning("Error: " + e.message)
-        gr.Warning("Error: " + str(e))        
-        gr.Info("Running out of HuggingFace GPU Quota? Try this demo hosted at UPenn.</br></br> https://ncut-pytorch.readthedocs.io/en/latest/demo/", duration=20)
+        # I assume this is a GPU quota error
         
+        # print(e)
+        # gr.Warning(e.message)
+        # if USE_HUGGINGFACE_ZEROGPU:
+        #     spaces.zero.gradio.HTMLError(e.message)
+        # gr.Warning("Error: " + e.message)
+        # gr.Warning("Error: " + str(e))        
+        info = 'Running out of HuggingFace GPU Quota? Please try <a style="white-space: nowrap;text-underline-offset: 2px;color: var(--body-text-color)" href="https://ncut-pytorch.readthedocs.io/en/latest/demo/">Demo hosted at UPenn</a>'
+        # gr.Info(info, duration=20)
+        message = e.message + "</br></br>---</br>" + info
+        raise gr.Error(message, duration=20)