alidev2002 commited on
Commit
a2ec9ec
Β·
verified Β·
1 Parent(s): 3af3d43

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -53,7 +53,10 @@ def generate(prompt):
53
  "-o", OUTPUT_IMAGE
54
  ]
55
 
56
- subprocess.run(cmd, check=True)
 
 
 
57
 
58
  return OUTPUT_IMAGE
59
 
 
53
  "-o", OUTPUT_IMAGE
54
  ]
55
 
56
+ env = os.environ.copy()
57
+ env["LD_LIBRARY_PATH"] = SD_DIR
58
+
59
+ subprocess.run(cmd, check=True, env=env)
60
 
61
  return OUTPUT_IMAGE
62