nikigoli commited on
Commit
2945802
1 Parent(s): 83303a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -41,6 +41,8 @@ def build_custom_prompter():
41
 
42
  build_custom_prompter()
43
  from gradio_image_prompter import ImagePrompter
 
 
44
  install_add_dependencies()
45
 
46
  class AppSteps(Enum):
 
41
 
42
  build_custom_prompter()
43
  from gradio_image_prompter import ImagePrompter
44
+ print("torch version")
45
+ print(torch.version.cuda)
46
  install_add_dependencies()
47
 
48
  class AppSteps(Enum):