Fabrice-TIERCELIN commited on
Commit
0dac2cb
1 Parent(s): cc91e24

if str(type(e)) == "'No GPU is currently available for you after 60s'":

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -139,10 +139,10 @@ def stage2_process(*args, **kwargs):
139
  return restore_in_Xmin(*args, **kwargs)
140
  except Exception as e:
141
  # NO_GPU_MESSAGE_INQUEUE
142
- print('Exception of type ' + str(type(e)))
143
  print("gradio.exceptions.Error: 'No GPU is currently available for you after 60s'")
144
- print('Exception of message ' + str(e))
145
- if str(e) == "'GPU task aborted'":
 
146
  print('Exception identified!!!')
147
  #if str(type(e)) == "<class 'gradio.exceptions.Error'>":
148
  #print('Exception of name ' + type(e).__name__)
 
139
  return restore_in_Xmin(*args, **kwargs)
140
  except Exception as e:
141
  # NO_GPU_MESSAGE_INQUEUE
 
142
  print("gradio.exceptions.Error: 'No GPU is currently available for you after 60s'")
143
+ print('str(type(e)) ' + str(type(e)))
144
+ print('str(e) ' + str(e))
145
+ if str(type(e)) == "'No GPU is currently available for you after 60s'":
146
  print('Exception identified!!!')
147
  #if str(type(e)) == "<class 'gradio.exceptions.Error'>":
148
  #print('Exception of name ' + type(e).__name__)