Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -53,7 +53,10 @@ def generate(prompt):
|
|
| 53 |
"-o", OUTPUT_IMAGE
|
| 54 |
]
|
| 55 |
|
| 56 |
-
|
|
|
|
|
|
|
|
|
|
| 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 |
|