basicv8vc commited on
Commit
f6daad4
1 Parent(s): 4cefdae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -25,8 +25,11 @@ def install_oneflow():
25
  )
26
 
27
 
28
- install_oneflow()
29
- import oneflow as flow
 
 
 
30
 
31
  ConstantLR_CODE = """oneflow.optim.lr_scheduler.ConstantLR(
32
  optimizer: Optimizer,
 
25
  )
26
 
27
 
28
+ try:
29
+ import oneflow as flow
30
+ except:
31
+ install_oneflow()
32
+ import oneflow as flow
33
 
34
  ConstantLR_CODE = """oneflow.optim.lr_scheduler.ConstantLR(
35
  optimizer: Optimizer,