Spaces:
Runtime error
Runtime error
lemonaddie
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -403,15 +403,10 @@ def run_demo_server(pipe):
|
|
403 |
|
404 |
def main():
|
405 |
|
406 |
-
if os.path.isdir(REPO_DIR):
|
407 |
-
shutil.rmtree(REPO_DIR)
|
408 |
-
repo = git.Repo.clone_from(REPO_URL, REPO_DIR)
|
409 |
-
repo.git.checkout(REPO_HASH)
|
410 |
-
|
411 |
sys.path.append(os.path.join(os.getcwd(), REPO_DIR))
|
412 |
|
413 |
from diffusers import DiffusionPipeline
|
414 |
-
pipeline = DiffusionPipeline.from_pretrained("
|
415 |
|
416 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
417 |
|
|
|
403 |
|
404 |
def main():
|
405 |
|
|
|
|
|
|
|
|
|
|
|
406 |
sys.path.append(os.path.join(os.getcwd(), REPO_DIR))
|
407 |
|
408 |
from diffusers import DiffusionPipeline
|
409 |
+
pipeline = DiffusionPipeline.from_pretrained("lemonaddie/Geowizard")
|
410 |
|
411 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
412 |
|