Exception identified!!!

app.py

@@ -138,12 +138,14 @@ def stage2_process(*args, **kwargs):
     try:
         return restore_in_Xmin(*args, **kwargs)
     except Exception as e:
+        # NO_GPU_MESSAGE_INQUEUE
         print('Exception of type ' + str(type(e)))
         print("gradio.exceptions.Error: 'No GPU is currently available for you after 60s'")
         print('Exception of message ' + str(e))
-        #print(NO_GPU_MESSAGE_INQUEUE)
-        if str(type(e)) == "<class 'gradio.exceptions.Error'>":
-            print('Exception of name ' + type(e).__name__)
+        if str(e) == "'GPU task aborted'":
+            print('Exception identified!!!')
+        #if str(type(e)) == "<class 'gradio.exceptions.Error'>":
+            #print('Exception of name ' + type(e).__name__)
         raise e
 
 def restore_in_Xmin(