pranavajay commited on
Commit
72135da
·
verified ·
1 Parent(s): 837ff1c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,13 +1,13 @@
1
  import gradio as gr
2
  import numpy as np
3
  import random
4
- from diffusers import KalpanaPipeline
5
  import torch
6
  import spaces
7
 
8
  device = "cuda" if torch.cuda.is_available() else "cpu"
9
 
10
- pipe = KalpanaPipeline.from_pretrained(
11
  "enhanceaiteam/kalpana",
12
  torch_dtype=torch.float16
13
  ).to(device)
 
1
  import gradio as gr
2
  import numpy as np
3
  import random
4
+ from diffusers import FluxPipeline
5
  import torch
6
  import spaces
7
 
8
  device = "cuda" if torch.cuda.is_available() else "cpu"
9
 
10
+ pipe = FluxPipeline.from_pretrained(
11
  "enhanceaiteam/kalpana",
12
  torch_dtype=torch.float16
13
  ).to(device)