Taejun Kim commited on
Commit
edc3710
1 Parent(s): 1c8e683
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -1,3 +1,7 @@
 
 
 
 
1
  import gradio as gr
2
  import os
3
  import allin1
@@ -48,6 +52,9 @@ CACHE_EXAMPLES = os.getenv('CACHE_EXAMPLES', '1') == '1'
48
 
49
 
50
  def analyze(path):
 
 
 
51
  path = Path(path)
52
  result = allin1.analyze(
53
  path,
 
1
+ import torch
2
+
3
+ print(f'=> is cuda available: {torch.cuda.is_available()}')
4
+
5
  import gradio as gr
6
  import os
7
  import allin1
 
52
 
53
 
54
  def analyze(path):
55
+ import torch
56
+ print(f'=> is cuda available: {torch.cuda.is_available()}')
57
+
58
  path = Path(path)
59
  result = allin1.analyze(
60
  path,