ehristoforu commited on
Commit
b506548
1 Parent(s): c413833

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,7 +14,7 @@ from diffusers import StableDiffusionPipeline
14
  DESCRIPTION = """
15
  # Fluently V1
16
 
17
- Model: [this](https://huggingface.co/ehristoforu/Fluently-v1)
18
  """
19
  if not torch.cuda.is_available():
20
  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")
29
 
30
  if torch.cuda.is_available():
31
  pipe = StableDiffusionPipeline.from_pretrained(
32
- "ehristoforu/Fluently-v1",
33
  torch_dtype=torch.float16,
34
  use_safetensors=True,
35
  )
 
14
  DESCRIPTION = """
15
  # Fluently V1
16
 
17
+ Model: [this](https://huggingface.co/ehristoforu/Fluently)
18
  """
19
  if not torch.cuda.is_available():
20
  DESCRIPTION += "\n<p>Running on CPU 🥶 This demo may not work on CPU.</p>"
 
29
 
30
  if torch.cuda.is_available():
31
  pipe = StableDiffusionPipeline.from_pretrained(
32
+ "ehristoforu/Fluently",
33
  torch_dtype=torch.float16,
34
  use_safetensors=True,
35
  )