Tonic commited on
Commit
cd2da70
β€’
1 Parent(s): 7de6816

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -1,12 +1,18 @@
1
  import gradio as gr
2
  from diffusers import StableDiffusionXLPipeline, DDIMScheduler
3
  import torch
4
- import sa_handler
5
- import inversion
6
  import numpy as np
7
  from diffusers.utils import load_image
8
  from PIL import Image
9
  import io
 
 
 
 
 
 
 
 
10
 
11
  # Model Load
12
  scheduler = DDIMScheduler(
 
1
  import gradio as gr
2
  from diffusers import StableDiffusionXLPipeline, DDIMScheduler
3
  import torch
 
 
4
  import numpy as np
5
  from diffusers.utils import load_image
6
  from PIL import Image
7
  import io
8
+ import sys
9
+ import os
10
+ current_dir = os.path.dirname(os.path.realpath(__file__))
11
+ if current_dir not in sys.path:
12
+ sys.path.append(current_dir)
13
+
14
+ import sa_handler
15
+ import inversion
16
 
17
  # Model Load
18
  scheduler = DDIMScheduler(