Update app.py

app.py

@@ -127,6 +127,12 @@ replace_grad = ReplaceGrad.apply
 import sys
 sys.path.append('./cloob-training')
 sys.path.append('./clip')
+# git isn't pulling the submodules for cloob-training so we need to add a path to clip
+with open('./cloob-training/cloob_training/model_pt.py', 'r+') as f:
+    content = f.read()
+    f.seek(0, 0)
+    f.write("import sys\n" + "sys.path.append('../clip')\n" + '\n' + content)
+        
 from cloob_training import model_pt, pretrained
 
 config = pretrained.get_config('cloob_laion_400m_vit_b_16_16_epochs')