lemonaddie commited on
Commit
af36d6b
1 Parent(s): ef5dcbb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -6
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("JUGGHM/temp_repo")
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