johnowhitaker commited on
Commit
4f44610
1 Parent(s): 639a6f3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -126,6 +126,7 @@ replace_grad = ReplaceGrad.apply
126
  #Load CLOOB model
127
  import sys
128
  sys.path.append('./cloob-training')
 
129
  from cloob_training import model_pt, pretrained
130
 
131
  config = pretrained.get_config('cloob_laion_400m_vit_b_16_16_epochs')
 
126
  #Load CLOOB model
127
  import sys
128
  sys.path.append('./cloob-training')
129
+ sys.path.append('./clip')
130
  from cloob_training import model_pt, pretrained
131
 
132
  config = pretrained.get_config('cloob_laion_400m_vit_b_16_16_epochs')