Spaces:

ehristoforu
/

Fluently

Paused

ehristoforu commited on Feb 20

Commit

b506548

•

1 Parent(s): c413833

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ from diffusers import StableDiffusionPipeline
 DESCRIPTION = """
 # Fluently V1
-Model: [this](https://huggingface.co/ehristoforu/Fluently-v1)
 """
 if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>Running on CPU 🥶 This demo may not work on CPU.</p>"
@@ -29,7 +29,7 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 if torch.cuda.is_available():
     pipe = StableDiffusionPipeline.from_pretrained(
-        "ehristoforu/Fluently-v1",
         torch_dtype=torch.float16,
         use_safetensors=True,
     )

 DESCRIPTION = """
 # Fluently V1
+Model: [this](https://huggingface.co/ehristoforu/Fluently)
 """
 if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>Running on CPU 🥶 This demo may not work on CPU.</p>"
 if torch.cuda.is_available():
     pipe = StableDiffusionPipeline.from_pretrained(
+        "ehristoforu/Fluently",
         torch_dtype=torch.float16,
         use_safetensors=True,
     )