fruitpicker01 commited on
Commit
acc030b
·
verified ·
1 Parent(s): 1fdf55c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -17,6 +17,12 @@ import string
17
  import io
18
  from datetime import datetime, timedelta
19
  import plotly.express as px
 
 
 
 
 
 
20
 
21
  morph = pymorphy2.MorphAnalyzer()
22
 
 
17
  import io
18
  from datetime import datetime, timedelta
19
  import plotly.express as px
20
+ import torch
21
+
22
+ if not torch.cuda.is_available():
23
+ raise RuntimeError("GPU not available. ZeroGPU initialization failed.")
24
+ else:
25
+ print("ZeroGPU initialized successfully.")
26
 
27
  morph = pymorphy2.MorphAnalyzer()
28